LeetCodePython566重塑矩阵详解

解题思路

本题的核心在于如何将一个给定的二维数组按照指定的新行数和列数进行重塑。具体步骤如下：

1. 计算原二维数组的总元素个数。
2. 检查新矩阵的总元素个数是否与原矩阵相等，如果不等则返回原矩阵。
3. 将原二维数组降为一维数组。
4. 根据新的行数和列数，重新构建矩阵。

代码实现

以下是Python代码实现：

class Solution(object):
    def matrixReshape(self, nums, r, c):
        ":type nums: List[List[int]]
        :type r: int
        :type c: int
        :rtype: List[List[int]]"
        # 初始化新矩阵
        new_matrix = [[0] * c for _ in range(r)]
        # 获取原矩阵的行数和列数
        original_rows = len(nums)
        original_cols = len(nums[0])
        # 检查是否可以重塑
        if original_rows * original_cols != r * c:
            return nums
        else:
            # 将原矩阵降为一维数组
            flat_list = []
            for row in nums:
                flat_list.extend(row)
            # 重新构建新矩阵
            for i in range(r):
                for j in range(c):
                    new_matrix[i][j] = flat_list[i * c + j]
            return new_matrix