这题难度不是太大,只要注意下标与结束的时候就可以了。
#566
#https://leetcode.com/problems/reshape-the-matrix/description/
#2018.4.29
class Solution(object):
def matrixReshape(self, nums, r, c):
i = 0
n = len(nums)
newlist = []
outs = []
for num in nums:
lens = len(num)
if n*lens != r*c:
return nums
for j in range (0,lens):
if i<c:
i+=1
newlist.append(num[j])
if i == c:
outs.append(newlist)
newlist= []
i=0
return outs
s = Solution()
nums = [
[1,2],
[3,4],
]
nlist = s.matrixReshape(nums,4,1)
print(nlist)