題目鏈接： https://leetcode-cn.com/problems/reshape-the-matrix

難度：簡單

通過率：61.6%

題目描述:

在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 都是正數。

思路:

思路一：遍歷所有

思路二：numpy 庫

代碼:

思路一：

class Solution:def matrixReshape(self, nums: List[List[int]], r: int, c: int) -> List[List[int]]:row, col = len(nums), len(nums[0])if row * col != r * c: return numsres, cur =[], []for num in itertools.chain.from_iterable(nums):cur.append(num)if len(cur) == c:res.append(cur)cur = []return res

思路二：

class Solution:def matrixReshape(self, nums: List[List[int]], r: int, c: int) -> List[List[int]]:import numpy as nprow, col = len(nums), len(nums[0])if row * col != r * c: return numsreturn np.reshape(nums, (r, c)).tolist()

總結

以上是生活随笔為你收集整理的遍历矩阵每一行穷举_[LeetCode] 566. 重塑矩阵的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。