【leetcode11-----重塑矩阵】

重塑矩阵

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

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

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

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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/reshape-the-matrix
 

题解一：

官解的解法十分简便，就是一个公式，把二维数组拉扁成一维数组，再由一维数组转化成r行c列的二维数组。由于是用行遍历顺序填充，行=n，列=m，把二维数组拉扁成一维数组时，二维数组中元素在一维数组中的下标(i,j)=n×i+j。把一维数组转化成r行c列的二维数组时，x为元素在一维数组中的下标，二维数组的行列下标分别i=x/r，j=x%r。代码如下：

 

class Solution {
    public int[][] matrixReshape(int[][] nums, int r, int c) {
        int m = nums.length;
        int n = nums[0].length;
        if (m * n != r * c) {
            return nums;
        }

        int[][] ans = new int[r][c];
        for (int x = 0; x < m * n; ++x) {
            ans[x / c][x % c] = nums[x / n][x % n];
        }
        return ans;
    }
}

题解二：

笔者自己写的效率非常低，当时没有想到官解的方法，所以用了很多循环，很麻烦，简单记录一下吧，代码如下：

class Solution {
    public int[][] matrixReshape(int[][] mat, int r, int c) {
        int rowlength=mat.length;//mat数组的行数
        int collength=mat[0].length;//mat数组的列数
        if(rowlength*collength!=r*c){
            return mat;
        }
        List<Integer> list=new ArrayList<Integer>();
        int[][] res=new int[r][c];

        //把mat数组按行遍历放到list集合中
        for(int i=0;i<rowlength;i++){
            for(int j=0;j<collength;j++){
                list.add(mat[i][j]);
            }
        }
        int n=0;
        //再把元素从集合中放到答案数组中
        for(int i=0;i<r;i++){
            for(int j=0;j<c;j++){
                res[i][j]=list.get(n);
                n++;
            }
        }
        return res;
    }
}