题目地址:
https://www.lintcode.com/problem/reshape-the-matrix/description
给定一个二维矩阵 A A A,再给定两个正整数 r r r和 c c c,要求将 A A A变换为 r r r行 c c c列的矩阵,并且保持行遍历相同。如果无法变换,则返回原矩阵,否则返回变换后的答案。
能变换当且仅当 A A A的元素个数等于 r c rc rc。判断完之后将原矩阵的数填到新矩阵里即可。代码如下:
public class Solution {
/**
* @param nums: List[List[int]]
* @param r: an integer
* @param c: an integer
* @return: return List[List[int]]
*/
public int[][] matrixReshape(int[][] nums, int r, int c) {
// write your code here
int m = nums.length, n = nums[0].length;
if (m * n != r * c) {
return nums;
}
int[][] res = new int[r][c];
for (int idx = 0, i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
res[idx / c][idx % c] = nums[i][j];
idx++;
}
}
return res;
}
}
时空复杂度 O ( m n ) O(mn) O(mn)。