演算法21----重塑矩陣 LeetCode566

標籤：否則   列表   style   遍曆   元素   ble   重點   迭代   too   

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. 給定矩陣的寬和高範圍在 [1, 100]。
2. 給定的 r 和 c 都是正數。

2、思路

先將原始矩陣轉化成一個迭代器，所有數都放進去，【也可以放進隊列中】，然後建立一個新的矩陣（大小為reshape的大小），一個一個往裡放數。

將列錶轉化成迭代器可以用itertools模組下chain.from_iterable(列表) 函數。然後用next函數一個一個取數

還有一個重點：建立二維矩陣的一種方法：

                for i in range(r):
                    result.append([])
                    for j in range(c):
                        result[i].append(next(chainnum))

第二種：result = [ [ ] * r for i in range(c) ] 【清單產生器】

3、代碼

from itertools import chainclass Solution(object):    def matrixReshape(self, nums, r, c):        """        :type nums: List[List[int]]        :type r: int        :type c: int        :rtype: List[List[int]]        """        if not nums:            return nums        elif not r or not c:            return nums        else:            h=len(nums)            l=len(nums[0])            if h*l != r*c:                return nums            else:                chainnum = chain.from_iterable(nums)                result = []                for i in range(r):                    result.append([])                    for j in range(c):                        result[i].append(next(chainnum))                return result

 

演算法21----重塑矩陣 LeetCode566