1605. 给定行和列的和求可行矩阵

发布时间 2023-04-08 23:03:27作者: lxy_cn

题目链接:1605. 给定行和列的和求可行矩阵

方法:贪心

解题思路

参考:思路?一个动画秒懂!附优化写法(Python/Java/C++/Go)

代码

class Solution {
public:
    vector<vector<int>> restoreMatrix(vector<int>& rowSum, vector<int>& colSum) {
        int m = rowSum.size(), n = colSum.size();
        vector<vector<int>> ans(m, vector<int>(n));
        for (int i = 0, j = 0; i < m, j < n; ) {
            int rs = rowSum[i], cs = colSum[j];
            if (rs < cs) {
                colSum[j] -= rs;
                ans[i ++ ][j] = rs;
            } else {
                rowSum[i] -=cs;
                ans[i][j ++ ] = cs;
            }
        }
        return ans; 
    }
};