问题描述:
将mn矩阵转换成rc矩阵。若尺寸支持转换,则返回转换后矩阵;若不支持,则返回原矩阵
思路:
先计算矩阵的长宽,判断是否尺寸一致。若一致,则建立新矩阵。将原矩阵的数字填入一个辅助序列,再将辅助序列填回新矩阵
代码如下:
class Solution {
public int[][] matrixReshape(int[][] mat, int r, int c) {
int height=mat.length;
int weight=mat[0].length;
if(r*c!=height*weight) return mat;
int[][] newMat= new int[r][c];
List<Integer> list=new ArrayList<>();
for(int i=0; i<height; i++){
for(int j=0; j<weight; j++){
list.add(mat[i][j]);
}
}
int k=0;
for(int i=0; i<r; i++){
for(int j=0; j<c; j++){
newMat[i][j]=list.get(k);
k++;
}
}
return newMat;
}
}
时间复杂度: O(mn)