class Solution {
public int[][] matrixReshape(int[][] mat, int r, int c) {
int m = mat.length;
int n = mat[0].length;
if (m * n != r * c) return mat;
int[] tmp = new int[m * n];
int[][] res = new int[r][c];
int k = 0;
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
tmp[k++] = mat[i][j];
}
}
k = 0;
for (int i = 0; i < r; i++) {
for (int j = 0; j < c; j++) {
res[i][j] = tmp[k++];
}
}
return res;
}
}
02-17
150
