566. 重塑矩阵

 

 

思路：将原nums中的元素都读出来，在往新list中加。

class Solution(object):
    def matrixReshape(self, nums, r, c):
        """
        :type nums: List[List[int]]
        :type r: int
        :type c: int
        :rtype: List[List[int]]
        """
        # nums行数
        r0 = len(nums)
        # nums列数
        c0 = len(nums[0])
        # 若reshape操作不合理，返回原list
        if r0 * c0 != r * c:
            return nums
        new = []
        for i in range(r0):
            for j in range(c0):
                new.append(nums[i][j])
        ans = []
        flag = 0
        for k in range(r):
            temp = []
            for h in range(c):
                temp.append(new[flag])
                flag += 1
            ans.append(temp)
        return ans


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