Krononberg

1605. Find Valid Matrix Given Row and Column Sums.cpp 본문

개발 로그/알고리즘

1605. Find Valid Matrix Given Row and Column Sums.cpp

k._. 2021. 10. 3. 15:30
class Solution {
public:

    vector<vector<int>> restoreMatrix(vector<int>& rowSum, vector<int>& colSum) {
        
        int m = rowSum.size();
        int n = colSum.size();
        
        vector<vector<int>>res(m,vector<int>(n));
        
        for(int i =0; i< m; i++){
            for(int j =0; j<n; j++){
                res[i][j] = min(rowSum[i],colSum[j]);
                rowSum[i] -= res[i][j];
                colSum[j] -= res[i][j];
            }
        }
        
        return res;
    }
};