刷题之旅从数组类型的题目开始。第二十五道题目是重塑矩阵，对应leetcode的题号为566。

image

题目描述

在MATLAB中，有一个非常有用的函数 reshape，它可以将一个矩阵重塑为另一个大小不同的新矩阵，但保留其原始数据。

给出一个由二维数组表示的矩阵，以及两个正整数r和c，分别表示想要的重构的矩阵的行数和列数。

重构后的矩阵需要将原始矩阵的所有元素以相同的行遍历顺序填充。

如果具有给定参数的reshape操作是可行且合理的，则输出新的重塑矩阵；否则，输出原始矩阵。

示例 1:

1
2
3
4
5
6
7
8
9
输入: 
nums = 
[[1,2],
 [3,4]]
r = 1, c = 4
输出: 
[[1,2,3,4]]
解释:
行遍历nums的结果是 [1,2,3,4]。新的矩阵是 1 * 4 矩阵, 用之前的元素值一行一行填充新矩阵。

示例 2:

1
2
3
4
5
6
7
8
9
10
输入: 
nums = 
[[1,2],
 [3,4]]
r = 2, c = 4
输出: 
[[1,2],
 [3,4]]
解释:
没有办法将 2 * 2 矩阵转化为 2 * 4 矩阵。 所以输出原矩阵。

注意：

  • 给定矩阵的宽和高范围在 [1, 100]。
  • 给定的 r 和 c 都是正数。

解题思路

基本思想就是遍历原数组，从上到下，从左到右遍历一遍，把所有元素逐个挪到一个新的数组中即可。因此要保证元素数量一样才能重塑，否则返回原数组。

那么我们只需要遍历原数组即可，一个一个地往新数组塞。

提交代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
class Solution {
    public int[][] matrixReshape(int[][] nums, int r, int c) {
        //判断是否能重塑
        if(r*c != nums.length * nums[0].length){
            return nums;
        }
        //定义一个新数组
        int[][] arr = new int[r][c];
        int row = 0;
        int col = 0;
        for(int i=0;i<nums.length;i++){
            for(int j=0;j<nums[0].length;j++){
                arr[row][col] = nums[i][j];
                //如果变量col遍历到了最后一列，那么就换行再去从头开始赋值
                if(col == c-1){
                    row++;
                    col = 0;
                }else{
                    col++;
                }
            }
        }
        return arr;
    }
}