題目

難度：★☆☆☆☆
類型：數(shù)組，幾何

在MATLAB中，有一個非常有用的函數(shù) reshape，它可以將一個矩陣重塑為另一個大小不同的新矩陣，但保留其原始數(shù)據(jù)。

給出一個由二維數(shù)組表示的矩陣，以及兩個正整數(shù)r和c，分別表示想要的重構(gòu)的矩陣的行數(shù)和列數(shù)。

重構(gòu)后的矩陣需要將原始矩陣的所有元素以相同的行遍歷順序填充。

如果具有給定參數(shù)的reshape操作是可行且合理的，則輸出新的重塑矩陣；否則，輸出原始矩陣。

注意
給定矩陣的寬和高范圍在 [1, 100]。
給定的 r 和 c 都是正數(shù)。

示例

示例 1:
輸入:
nums =
[[1,2],
[3,4]]
r = 1, c = 4
輸出:
[[1,2,3,4]]
解釋:
行遍歷nums的結(jié)果是 [1,2,3,4]。新的矩陣是 1 * 4 矩陣, 用之前的元素值一行一行填充新矩陣。
示例 2:
輸入:
nums =
[[1,2],
[3,4]]
r = 2, c = 4
輸出:
[[1,2],
[3,4]]
解釋:
沒有辦法將 2 * 2 矩陣轉(zhuǎn)化為 2 * 4 矩陣。 所以輸出原矩陣。

解答

該函數(shù)也相當(dāng)于python庫numpy中矩陣的reshape方法，reshape的最基本的原理就是把高維矩陣?yán)梢涣袛?shù)字，然后按照新的方法重構(gòu)出不同形狀的矩陣。但是新矩陣中元素個數(shù)需要和原矩陣一致。

1. 判斷新矩陣的維度是否滿足要求，如果無法重構(gòu)則直接返回原矩陣；

2. 遍歷原矩陣中每一個數(shù)字，把數(shù)字逐一裝填進(jìn)臨時列表row_中，如果該列表長度達(dá)到了指定寬度，那么把該列表作為一行加入到最終結(jié)果矩陣nums_中；

3. 所有數(shù)字遍歷結(jié)束后，返回重構(gòu)矩陣即可。

class Solution:
    def matrixReshape(self, nums, r, c):
        if r * c != len(nums) * len(nums[0]):   # 如果重構(gòu)矩陣的面積與原始矩陣不相等
            return nums                         # 無法進(jìn)行重構(gòu)，返回原始矩陣

        nums_, row_, c_ = [], [], 0             # 初始化重構(gòu)后數(shù)組，重構(gòu)時的一行以及該行中元素個數(shù)

        for row in nums:                        # 先遍歷行
            for num in row:                     # 嵌套遍歷列
                row_.append(num)                # 取出元素放在臨時列表中
                c_ += 1                         # 該列表元素數(shù)+1
                if c_ == c:                     # 如果列表中的元素個數(shù)達(dá)到了列數(shù)
                    nums_.append(row_)          # 那么把這個列表放在重構(gòu)數(shù)組中作為一行
                    row_, c_ = [], 0            # 清零臨時列表和列表中元素個數(shù)

        return nums_                            # 返回結(jié)果

如有疑問或建議，歡迎評論區(qū)留言~