b_lc_给定行和列的和求可行矩阵(找行和列的和的最小值填)
Posted wdt1
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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+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_给定行和列的和求可行矩阵(找行和列的和的最小值填)的主要内容,如果未能解决你的问题,请参考以下文章
在 pandas 中,如何在具有匹配行和列的 3 个单独数据帧之间建立相关矩阵?
2021-10-01:矩阵置零。给定一个 m x n 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。进阶:一个直观的解决方案是使用 O(mn) 的额外空间(