[LeetCode]566. 重塑矩阵(java实现)
1. 题目
2. 读题(需要重点注意的东西)
思路:拉伸二维数组为一维,然后将该数组赋值给新的矩阵
3. 解法
解法:
class Solution {
public int[][] matrixReshape(int[][] mat, int r, int c) {
if(mat.length*mat[0].length != r*c){
return mat;
}
int n = 0;
int[] nums = new int[mat.length*mat[0].length];
for(int i = 0; i < mat.length ; i++){
for(int j = 0; j < mat[0].length; j++){
nums[n++] = mat[i][j];
}
}
n = 0;
int[][] new_mat = new int[r][c];
for(int i = 0; i < r; i++){
for(int j = 0; j < c; j++){
new_mat[i][j] = nums[n++];
}
}
return new_mat;
}
}