forked from larissalages/code_problems
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path1605.cpp
More file actions
27 lines (24 loc) · 897 Bytes
/
1605.cpp
File metadata and controls
27 lines (24 loc) · 897 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
class Solution {
public:
void findElements(int i,int j, vector<vector<int>>& ans, vector<int>& rowSum, vector<int>& colSum){
if(i>=0 and j>=0 and i<rowSum.size() and j<colSum.size() and ans[i][j]==-1){
ans[i][j]=min(rowSum[i],colSum[j]);
rowSum[i]-=ans[i][j];
colSum[j]-=ans[i][j];
findElements(i-1,j,ans,rowSum,colSum);
findElements(i,j-1,ans,rowSum,colSum);
}
return;
}
vector<vector<int>> restoreMatrix(vector<int>& rowSum, vector<int>& colSum) {
vector<vector<int>>ans(rowSum.size(),vector<int>(colSum.size(),-1));
for(int i=rowSum.size()-1;i>=0;i--){
for(int j=colSum.size()-1;j>=0;j--){
if(ans[i][j]==-1){
findElements(i,j,ans,rowSum,colSum);
}
}
}
return ans;
}
};