【LeetCode】Day94-重塑矩阵

最新推荐文章于 2023-01-21 19:00:36 发布

倒过来是圈圈

最新推荐文章于 2023-01-21 19:00:36 发布

阅读量119

点赞数

分类专栏： Leetcode进阶之路文章标签： leetcode 矩阵算法

本文链接：https://blog.csdn.net/qq_43417265/article/details/125582948

版权

Leetcode进阶之路专栏收录该内容

232 篇文章 1 订阅

订阅专栏

题目

566. 重塑矩阵【简单】

题解

二维数组的一维表示

题目需要我们做的事情相当于：

将二维数组 nums 映射成一个一维数组；
将这个一维数组映射回 r 行 c 列的二维数组。

具体原理是：对于 x∈[0,mn)，第 x 个元素在 nums 中对应的下标为 (x / n,x % n)，而在新的重塑矩阵中对应的下标为 (x / c,x % c)。所以我们在代码中做对应映射即可。

class Solution {
    public int[][] matrixReshape(int[][] mat, int r, int c) {
        int m=mat.length,n=mat[0].length;
        if(m*n!=r*c)
            return mat;
        
        int[][] newMat=new int[r][c];
        for(int num=0;num<m*n;num++){
            newMat[num/c][num%c]=mat[num/n][num%n];
        }
        return newMat;
    }
}