力扣:566.重塑矩阵

98 阅读1分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第6天,点击查看活动详情

描述

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

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

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

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

  • 示例 1:
输入:mat = [[1,2],[3,4]], r = 1, c = 4
输出:[[1,2,3,4]]
  • 示例 2:
输入:mat = [[1,2],[3,4]], r = 2, c = 4
输出:[[1,2],[3,4]]
  • 提示:
  • m == mat.length
  • n == mat[i].length
  • 1 <= m, n <= 100
  • -1000 <= mati <= 1000
  • 1 <= r, c <= 300

解析

根据题意,看了官方题解和一些大神的思路,谈谈对本题的理解。

  • 首先计算原二维数组的长度,和每一位元素长度,他们的乘积决定了自定义的两个变量是否可以实现要求

  • 根据输入的值乘积是否等于原本的行列乘积,不符合要求原样返回

  • 如果符合要求,先根据自定义两个变量,根据入参来声明二维数组。

  • 通过for循环,n2控制mat的取值范围,声明初始值为0的两个指针,用n2约束二维数组中一位数组的长度。

    Java解题过程如下:

class Solution {
    public int[][] matrixReshape(int[][] mat, int r, int c) {
        //初始化行、列
        int n1 = mat.length;
        int n2 = mat[0].length;
        // 判断乘积和原本的行列乘积,则说明无效
        if (n1 * n2 != r * c) {
            return mat;
        }
        int[][] d = new int[r][c];
        int x1 = 0;
        int x2 = 0;
        for (int i = 0; i < r; i++) {
            for (int j = 0; j < c; j++) {
                if (x2 == n2) {
                    x1 += 1;
                    x2 = 0;
                }
                int index1 = x1;
                int index2 = x2;
                d[i][j] = mat[index1][index2];
                x2++;
            }
        }
        return d;
    }
}

运行结果:

执行结果:通过

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

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