数组与矩阵 - 改变矩阵维度

最新推荐文章于 2023-06-01 21:26:30 发布

原创最新推荐文章于 2023-06-01 21:26:30 发布 · 325 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #数据结构

LeetCode 专栏收录该内容

78 篇文章

订阅专栏

本文详细介绍了LeetCode中566题——重塑矩阵的解题思路，提供了一种直接法的Java实现。该方法首先检查原矩阵的元素数量是否能重塑为给定的新尺寸，若可以则进行重塑，否则返回原矩阵。时间复杂度为O(rc)，空间复杂度为O(1)。

1.题目

LeetCode: 566. 重塑矩阵

【easy】

2.解题

方法一：直接法

获取nums的长h和宽w，如果h * w != r * c，则直接返回nums，否则将原数组元素转移到新数组，即n%w为行的个数，n/w为列的个数。

java：

class Solution {
    public int[][] matrixReshape(int[][] nums, int r, int c) {
        int h = nums.length;
        int w = nums[0].length;
        if (h * w != r * c) return nums;
        int[][] res = new int[r][c];
        int n = 0;
        for (int i = 0; i < r; i++) {
            for (int j = 0; j < c; j++) {
                res[i][j] = nums[n / w][n % w];
                n++;
            }
        }
        return res;
    }
}