leetcode练习（566）

题566

题目要求是实现一个reshape函数，输入一个二维数组，以及重组后的行和列。输出是将该二维数组整形变成指定行列的形式，如果行列之积不等于数组元素个数，就返回原数组。
我的思路是先将二维数组变为一维数组，先判断给定的行列能否满足重组数组的要求，若不能就直接返回原数组。若符合，就遍历行和列，依次取数，先取列数，获得n；再遍历行，res.append(n)，结果就返回res。
代码如下：

class Solution(object):
    def matrixReshape(self, nums, r, c):

        """
        :type nums: List[List[int]]
        :type r: int
        :type c: int
        :rtype: List[List[int]]
        """
        re = []  # 将二维数组转化为一维
        res = [] # 保存分类结果
        for k in nums[0]:
            re.append(k)
        for k in nums[1]:
            re.append(k)
        if r * c == len(re): # 如果行列数符合要求
            for j in range(r):
                n = []
                for i in range(c):
                     n.append(re[j*c + i])
                res.append(n)
            return res
        else:    # 如果行列数不符合要求就返回原数组
            return nums

nums = [[1,2],[3,4]]
s = Solution()
print(s.matrixReshape(nums,2,4))

结果如下：