1170. 重塑矩阵

class Solution:
    """
    @param nums: List[List[int]]
    @param r: an integer
    @param c: an integer
    @return: return List[List[int]]
    """
    ‘‘‘
    大致思路:
    1.首先，初始化res = [],l = [],得到nums的维度，即len(nums)和len(nums[0])，和r*c进行判断，如果乘积>=r*c的话，则说明可以重塑矩阵，否则输入原矩阵。
    2.先将nums拼接成一维数组，即l = [...]的形式,
    3.如果可以重塑矩阵的话，根据给出的r和c，进行循环l，给出current_dic = [],如果长度为c的话，则append到res里面，一直到res长度为r的时候break，最后返回res
    ‘‘‘
    def matrixReshape(self,nums,r,c):
        res = []
        current_dic = []
        l = []

        #如果不符合条件的话
        if len(nums[0])*len(nums) < r*c:
            return  nums
        
        for i in nums:
            l.extend(i)

        #否则
        for column in l:
            current_dic.append(column)
            if len(current_dic) == c:
                res.append(current_dic)
                current_dic = []
            if len(res) == r:
                return res