[LeetCode Python3]566. Reshape the Matrix(重塑矩阵)

输入: 
nums = 
[[1,2],
 [3,4]]
r = 1, c = 4
输出: 
[[1,2,3,4]]
解释:
行遍历nums的结果是 [1,2,3,4]。新的矩阵是 1 * 4 矩阵, 用之前的元素值一行一行填充新矩阵。

输入: 
nums = 
[[1,2],
 [3,4]]
r = 2, c = 4
输出: 
[[1,2],
 [3,4]]
解释:
没有办法将 2 * 2 矩阵转化为 2 * 4 矩阵。 所以输出原矩阵。

class Solution:
    def matrixReshape(self, nums, r, c):
        """
        :type nums: List[List[int]]
        :type r: int
        :type c: int
        :rtype: List[List[int]]
        """
        rr = len(nums)
        cc = len(nums[0])
        retl = []
        if rr * cc < r * c:
            return nums
        else:
            t = []
            for i in range(rr):
                for j in range(cc):
                    t.append(nums[i][j])
            m = 0
            for i in range(r):
                retl.append(t[m:m + c])
                m += c
        return retl