nums =[[1,2],[3,4]]
r =1
c =4
matrixReshape(nums, r, c)
输出:
[[1,2,3,4]]
示例2:
nums =[[1,2],[3,4]]
r =2
c =4
matrixReshape(nums, r, c)
输出:
[[1,2],[3,4]]
时间复杂度
该函数的时间复杂度为 $O(m \times n)$,m 和 n 分别为原始矩阵的行数和列数。
代码实现
from typing import List
defmatrixReshape(nums: List[List[int]], r:int, c:int)-> List[List[int]]:
m, n =len(nums),len(nums[0])if r*c != m*n:return nums
res =[[0]* c for _ inrange(r)]for i inrange(r*c):
res[i//c][i%c]= nums[i//n][i%n]return res