题意:将数组转换为指定行数列数的数组。
用python刷leetcode真的是在作弊。不过人生苦短~
思路:将原数组转为一位数组,在利用分片截取list。
class Solution(object):
def matrixReshape(self, nums, r, c):
l = len(nums) * len(nums[0])
if l < r*c :
return nums
else :
rs = []
rs2 = []
for li in nums:
rs += li
for i in range(0, r) :
end = c*(i + 1)
start = c*i
rs2.append(rs[start:end:])
return rs2