【力扣】566.重塑矩阵

最新推荐文章于 2022-05-25 08:57:27 发布

彩云若锦

最新推荐文章于 2022-05-25 08:57:27 发布

阅读量183

点赞数 1

分类专栏： leetcode题解文章标签：矩阵 leetcode 算法

本文链接：https://blog.youkuaiyun.com/m0_60625485/article/details/124651604

版权

leetcode题解专栏收录该内容

11 篇文章

订阅专栏

原题：

给你一个由二维数组 mat表示的m x n矩阵，以及两个正整数r和c，分别表示想要的重构的矩阵的行数和列数。
重构后的矩阵需要将原始矩阵的所有元素以相同的行遍历顺序填充。
如果具有给定参数的 reshape 操作是可行且合理的，则输出新的重塑矩阵；否则，输出原始矩阵。

解题思路

设为 $r e s u l t [r] [c]$ 要返回的值，对于 $r e s u l t$ 中每个元素遍历并赋值，注意处理原数组中脚标的变化即可。

代码

class Solution {
    public int[][] matrixReshape(int[][] mat, int r, int c) {
        int m = mat.length;
        int n = mat[0].length;
        int i,j;
        int[][] result = new int[r][c];
        int k = 0,l = 0;
        if(m*n != r*c) return mat;
        for(i = 0;i < r; i++){
            for(j = 0;j < c; j++,l++){
                if(l == n){
                    l = 0;
                    k++;
                }
                result[i][j] = mat[k][l];
            }
        }
        return result;
    } 
}