matrixReshape-重塑矩阵

题意

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

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

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

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

示例 1:

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

示例 2:

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

注意:

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

解题思路

此题简单题,一些思路放入代码中,代码中含文字说明

代码演示

class Solution {
    
    
    public int[][] matrixReshape(int[][] nums, int r, int c) {
    
    
            int han=nums.length;
            int lie=nums[0].length;
            //不满足条件,不可以reshape的情况
            if(han*lie!=r*c)
                return nums;
            //重新定义一个二维数组,用于存放reshape的值
            int[][] reshape=new int[r][c];
            //定义两个变量,rr代表reshape中的行,rc代表reshape中的列,初始值都是0
            int rr=0,rc=0;
            //双重循环,从nums中取数
            for (int i=0;i<han;i++)
            {
    
    
                for (int j=0;j<lie;j++)
                {
    
    
                    //将nums中的数取出来,按照要求放入到reshape中
                    reshape[rr][rc]=nums[i][j];
                    //以行的方式存储,reshape当前行右移
                    rc++;
                    //当一行满的时候,列向下移,当前行的指针重新归0
                    if(rc==c)
                    {
    
    
                        rr++;
                        rc=0;
                    }
                }
            }

            return reshape;
    }
}

效果

info
解答成功:
执行耗时:1 ms,击败了100.00% 的Java用户
内存消耗:39.7 MB,击败了25.85% 的Java用户

猜你喜欢

转载自blog.csdn.net/tangshuai96/article/details/113832105