获取动态二维数组元素的行/列

Question

我目前使用以下代码来获取2D int数组中元素的当前行和列:

const int num_rows = 5;
const int num_columns = 7;
int a[num_rows][num_columns];
int *p = &a[2][4];
int row = (p - &a[0][0]) / num_columns;
int col = (p - &a[row][0]);

这工作得很好，但现在我需要改变代码，以采取行和列的数量作为参数。据我所知，这意味着我需要动态地创建2D数组:

int** ary = new int*[sizeX]; 
for(int i = 0; i < sizeX; ++i) 
    ary[i] = new int[sizeY]; 

如果我以这种方式创建2D数组，上面的代码将查找行/列中断。我能做什么?

Answer 1

 int *ary = new int [sizeX * sizeY]; // allocate memory for the 2d array
 for(int i = 0; i < sizeX; ++I)
     for(j = 0; j < sizeY; ++j) 
          ary[i * sizeY + j] = 0;
  // to get the value of collumn c and row r: use ary[c * sizeY + r];

Answer 2

int row = (p - &a[0][0]) / num_columns;
int col = (p - &a[row][0]);

远离指针运算是个好主意。别这样!你实际上是减去指针，然后除以num_columns，这个数字将是随机的。

如果您想获得一个元素的行/列，请一次搜索一个元素。

for(int row=0; row<num_rows; ++row) {
   for(int column=0; column<num_columns; ++column) {
      if (a[row][column] == element) {
         // we got it!
      }
   }
}