首页 > 其他 > 详细

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

时间:2020-10-04 09:58:37      阅读:47      评论:0      收藏:0      [点我收藏+]

给定 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+m; i++) {
            int x=-1, y=-1;
            for (int r=0; r<n; r++) if (rowSum[r]!=0 && (x==-1 || rowSum[r]<rowSum[x])) x=r;
            for (int c=0; c<m; c++) if (colSum[c]!=0 && (y==-1 || colSum[c]<colSum[y])) y=c;
            if (x==-1 || y==-1) break;
            g[x][y]=min(rowSum[x], colSum[y]);
            rowSum[x]-=g[x][y];
            colSum[y]-=g[x][y];
        }
        return g;
    }
};

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

原文:https://www.cnblogs.com/wdt1/p/13766409.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!