在2D数组中创建类似元素的集合

I am trying to solve a problem that is based on a 2D array. This array contains different kinds of elements (from a total of 3 possible kinds). Lets assume the kind as X, Y, Z.I am trying to solve a problem that is based on