题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))
结果如下: