b_lc_给定行和列的和求可行矩阵(找行和列的和的最小值填)

给定 rowSum[i] 是二维矩阵中第 i 行元素的和, colSum[j] 是第 j 列元素的和
请找到大小为 rowSum.length x colSum.length 的任意 非负整数 矩阵,且该矩阵满足 rowSum 和 colSum 的要求。

思路
我是这样蒙的:越先找到越小/越大的值填了的容错能力更强,

class Solution {
public:
    vector<vector<int>> restoreMatrix(vector<int>& rowSum, vector<int>& colSum) {
        int n=rowSum.size(), m=colSum.size();
        vector<vector<int>> g(n, vector<int>(m));
        for (int i=0; i<n; i++)
        for (int j=0; j<m; j++) {
            g[i][j]=min(rowSum[i], colSum[j]);
            rowSum[i]-=g[i][j];
            colSum[j]-=g[i][j];
        }
        return g;
    }
};
posted @ 2020-10-04 09:02  童年の波鞋  阅读(169)  评论(0编辑  收藏  举报