每日leetcode-数组-566. 重塑矩阵

分类：数组-二维数组变换

题目描述：

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

给出一个由二维数组表示的矩阵，以及两个正整数r和c，分别表示想要的重构的矩阵的行数和列数。

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

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

解题思路：

class Solution:
    def matrixReshape(self, mat: List[List[int]], r: int, c: int) -> List[List[int]]:
        m = len(mat)  # 行数
        n = len(mat[0]) # 列数
        if m * n != r * c:
            return mat
        newmat = [[0]*c for _ in range(r)]
        for x in range(m*n):
            newmat[x // c][x % c] = mat[x // n][x % n]
        return newmat

时间复杂度：O(rc)。

空间复杂度：O(1)。这里的空间复杂度不包含返回的重塑矩阵需要的空间。

 

解题思路2：

按行列遍历每个元素，重新c个元素成一行，一共又r行即可。

class Solution:
    def matrixReshape(self, mat: List[List[int]], r: int, c: int) -> List[List[int]]:
        m = len(mat)  # 行数
        n = len(mat[0]) # 列数
        if m * n != r * c:
            return mat
        result = []
        line = []
        for i in range(m):
            for j in range(n):
                line.append(mat[i][j])
                if len(line) == c:
                    result.append(line)
                    line = []
        return result