数据结构-566. 重塑矩阵

127 阅读2分钟

Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情

一、题目描述:

在 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

来源:力扣(LeetCode) 链接:leetcode-cn.com/problems/re… 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

二、思路分析:

题目分析

因为是把矩阵转换成一个面积相同长宽不同的新矩阵,所以要判断的第一步是新要求的R*C是否和原来的矩阵面积相同,如果不同就直接返回如果相同,说明符合条件就可以开始分配空间准备进行数据转移

解题过程如下: 新建一个矩阵作为返回值 遍历 r* c 个元素的并找到两个矩阵间对应的位置。

之后巧用/和%运算 /列数-->可以取到每一行 %列数->可以取到每一列 两个搭配使用即可

三、AC 代码:

class Solution {
  public int[][] matrixReshape(int[][] nums, int r, int c) {
    int row = nums.length, column = nums[0].length;
    if (nums.length == 0 || row * column != r * c) 
        return nums;
    int[][] result = new int[r][c];
    for (int i = 0;i < r * c; i++) 
        result[i / c][i % c] = nums[i / column][i % column];
    return result;        
}
​
}

四、总结:

image.png

掘友们,解题不易,留下个赞或评论再走吧!谢啦~ 💐