bumpsoo 2024-07-20 23:01:59 +09:00
parent 6669e03a5e
commit e46461e1e1

View file

@ -0,0 +1,17 @@
# https://leetcode.com/problems/find-valid-matrix-given-row-and-column-sums
from typing import List
class Solution:
def restoreMatrix(self, rowSum: List[int], colSum: List[int]) -> List[List[int]]:
ret: List[List[int]] = []
i = 0
while i < len(rowSum):
ret.append([])
j = 0
while j < len(colSum):
ret[i].append(min(rowSum[i], colSum[j]))
rowSum[i] -= ret[i][j]
colSum[j] -= ret[i][j]
j += 1
i += 1
return ret