【LeetCode】Day94-重塑矩阵

题目

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;
    }
}

时间复杂度：$O(rc)$

空间复杂度：$O(1)$