对于m*n大小的数组a,第i个元素的位置为a[i/n][i%n] (i/n为所在行,i%n为所在列)
class Solution {
public int[][] matrixReshape(int[][] mat, int r, int c) {
int m = mat.length, n = mat[0].length;
if(m*n != r*c) return mat;
int[][] result = new int[r][c];
for(int i = 0; i < m*n; i++){
result[i/c][i%c] = mat[i/n][i%n];
}
return result;
}
}