leetcode如何重塑矩陣

這篇文章主要為大家展示了“leetcode如何重塑矩陣”，內容簡而易懂，條理清晰，希望能夠幫助大家解決疑惑，下面讓小編帶領大家一起研究并學習一下“leetcode如何重塑矩陣”這篇文章吧。

一、題目內容

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

二、解題思路

直接按照整除列數得到行數，取余得到每行的位置，逐一賦值即可。

三、代碼

class Solution:
    def matrixReshape(self, nums: list, r: int, c: int) -> list:
        if len(nums) * len(nums[0]) != r * c:
            return nums

        ans = [[0 for _ in range(c)] for _ in range(r)]
        for i in range(len(nums) * len(nums[0])):
            ans[i // c][i % c] = nums[i // len(nums[0])][i % len(nums[0])]

        return ans


if __name__ == '__main__':
    s = Solution()
    nums = [[1, 2],
            [3, 4]]
    r = 1
    c = 4
    ans = s.matrixReshape(nums, r, c)
    print(ans)

以上是“leetcode如何重塑矩陣”這篇文章的所有內容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內容對大家有所幫助，如果還想學習更多知識，歡迎關注億速云行業資訊頻道！