566 重塑矩阵（模拟）

左手的ㄟ右手 2022-08-31 09:54 3阅读 0赞

**1. 问题描述：**

在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 都是正数。  
来源：力扣（LeetCode）  
链接：https://leetcode-cn.com/problems/reshape-the-matrix

**2. 思路分析：**

分析题目可以知道我们模拟整个过程即可，我们可以先从前往后遍历一下原数组，将当前原数组的位置映射到一维数组按行展开的位置，然后将其映射到重构之后的新数组的对应位置即可。

**3. 代码如下：**

from typing import List
    
    
    class Solution:
        def matrixReshape(self, mat: List[List[int]], r: int, c: int) -> List[List[int]]:
            n, m = len(mat), len(mat[0])
            # 只有两个矩阵的元素相等的时候才可以重构
            if r * c != n * m: return mat
            res = [[0] * c for i in range(r)]
            # 遍历的时候先计算出当前的mat[i][j]对应的一维数组的位置, 然后再映射到重构之后的二维数组的位置
            for i in range(n):
                for j in range(m):
                    k = i * m + j
                    res[k // c][k % c] = mat[i][j]
            return res