例题
力扣1605:
解析
构造空矩阵,然后依次遍历矩阵元素,在(i,j)位置处选择rowSum[i]和colSum[j]中较小的值x填入,之后rowSum[i]和colSum[j]都减去x,继续遍历其他位置。
代码
class Solution:
def restoreMatrix(self, rowSum: List[int], colSum: List[int]) -> List[List[int]]:
m, n = len(rowSum), len(colSum)
ans = [[0] * n for _ in range(m)]
for i in range(m):
for j in range(n):
x = min(rowSum[i], colSum[j])
ans[i][j] = x
rowSum[i] -= x
colSum[j] -= x
return ans
Comments NOTHING