[数组][简单]566.重塑矩阵

---

tags:

• 简单
• 参考
• 数组
  flag: green
  style: summer
  data: ‘2019-6-14’

---

566.重塑矩阵

一、问题

在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 row = nums.length;
        if (row == 0) {
            return nums;
        }
        int col = nums[0].length;
        if (row * col != r * c) {
            return nums;
        }
        
        int total = r * c;
        int[][] res = new int[r][c];
        for(int t = 0; t < total; t++) {            
            res[t/c][t%c] = nums[t/col][t%col];
        }
        
        return res;
    }
}

执行用时 :4 ms, 在所有Java提交中击败了52.34%的用户

内存消耗 :56.5 MB, 在所有Java提交中击败了7.85%的用户

• 自己答案

class Solution {
    public int[][] matrixReshape(int[][] nums, int r, int c) {
        
        //新建一个转换之后的数组
        int[][] ans = new int[r][c];
        //判断能不能转换，行列坐标乘积前后是否相等
        int rows = nums.length;
        int columns = nums[0].length;
        if (rows * columns != r * c || nums == null) {
            return nums;
        }else{   
            //循环原来的数组，当原来元素下标大于r之后，行坐标加一       
            int m = 0, n = 0;
            for (int i = 0;i < rows;i++ ) {
                for (int j = 0; j < columns; j++) {  
                // 如果列等于给定的列数目，行数+1    
                    if (n == c) { 
                       m++;
                       n = 0;
                    }
                     ans[m][n] = nums[i][j];
                     n++;
                }
            }
        }

        return ans;
    }   
}

执行用时 :3 ms, 在所有Java提交中击败了93.40%的用户

内存消耗 :49.9 MB, 在所有Java提交中击败了49.68%的用户

总结：

• 一定要对输入矩阵的合法性进行判读
• 写代码小错误一定要注意