https://leetcode-cn.com/problems/loud-and-rich/description/
在一组 N 个人(编号为 0, 1, 2, ..., N-1
)中,每个人都有不同数目的钱,以及不同程度的安静(quietness)。
为了方便起见,我们将编号为 x
的人简称为 "person x
"。
如果能够肯定 person x
比 person y
更有钱的话,我们会说 richer[i] = [x, y]
。注意 richer
可能只是有效观察的一个子集。
另外,如果 person x
的安静程度为 q
,我们会说 quiet[x] = q
。
现在,返回答案 answer
,其中 answer[x] = y
的前提是,在所有拥有的钱不少于 person x
的人中,person y
是最安静的人(也就是安静值 quiet[y]
最小的人)。
示例:
输入:richer = [[1,0],[2,1],[3,1],[3,7],[4,3],[5,3],[6,3]], quiet = [3,2,5,4,6,1,7,0]
输出:[5,5,2,5,4,5,6,7]
解释:
answer[0] = 5,
person 5 比 person 3 有更多的钱,person 3 比 person 1 有更多的钱,person 1 比 person 0 有更多的钱。
唯一较为安静(有较低的安静值 quiet[x])的人是 person 7,
但是目前还不清楚他是否比 person 0 更有钱。
answer[7] = 7,
在所有拥有的钱肯定不少于 person 7 的人中(这可能包括 person 3,4,5,6 以及 7),
最安静(有较低安静值 quiet[x])的人是 person 7。
其他的答案也可以用类似的推理来解释。
提示:
1 <= quiet.length = N <= 500
0 <= quiet[i] < N
,所有quiet[i]
都不相同。0 <= richer.length <= N * (N-1) / 2
0 <= richer[i][j] < N
richer[i][0] != richer[i][1]
richer[i]
都是不同的。- 对
richer
的观察在逻辑上是一致的。
通过了82/86个测试案例,算法正确,但是效率出问题了。
vector<int> loudAndRich(vector<vector<int>>& richer, vector<int>& quiet) {
vector< vector<int> > vr;
vector<int> res;
for (int i = 0; i<quiet.size(); i++) {
vector<int> si;
si.push_back(i);
vr.push_back(si);
}
for (int i = 0; i<richer.size(); i++) {
vector<int> vi = richer[i];
int a = vi[0], b = vi[1];
for (int j = 0; j<vr[a].size(); j++) {
if (find(vr[b].begin(), vr[b].end(), vr[a][j]) == vr[b].end())
vr[b].push_back(vr[a][j]);
}
//set_union(vr[a].begin(),vr[a].end(),vr[b].begin(),.end(),inserter(vr[b],.begin()));
dfs(vr, b, a);
}
for (int i = 0; i<quiet.size(); i++) {
int min = 10000, l = 0;
vector<int> si = vr[i];
vector<int>::iterator it;
for (it = si.begin(); it != si.end(); it++) {
int t = (*it);
//cout<<t;
if (quiet[t]<min) {
min = quiet[t];
l = t;
}
}
res.push_back(l);
//cout<<endl;
}
return res;
}
void dfs(vector< vector<int> > &vr, int b, int a) { //遍历集合,找到 b,添加 a
//cout<<b<<":"<<a<<endl;
for (int j = 0; j<vr.size(); j++) {
vector<int> sii = vr[j];
if (find(sii.begin(), sii.end(), b) != sii.end()) { //sii.count(b))
if (find(sii.begin(), sii.end(), a) == sii.end()) {
vr[j].push_back(a);
dfs(vr, j, a);
}
}
}
}
思路2 利用图的思想。
class Solution {
public:
int n;
vector<int> revale,qp,qv;
vector<int> loudAndRich(vector<vector<int>>& richer, vector<int>& quiet) {
n=quiet.size();
qp.resize(n);
qv.resize(n);
vector<vector<int>> res;
res.resize(n);
//返回值,最安静的人和值
vector<bool> vist(n,false);
for(int i=0; i<richer.size(); i++) {
int a=richer[i][0];
int b=richer[i][1];
res[b].push_back(a);
}
for(int i=0; i<n; i++) {
dfs(res,i,n,vist,quiet);
revale.push_back(qp[i]);
}
return revale;
}
void dfs( vector<vector<int>> & res, int i,int n,vector<bool> & vist,vector<int>& quiet) {
if(vist[i]) return;
int imin=quiet[i],p=i;
for(int j=0; j<res[i].size(); j++) {
if(!vist[res[i][j]])
dfs(res,res[i][j],n,vist,quiet);
if(imin>qv[res[i][j]]) {
imin=qv[res[i][j]];
p=qp[res[i][j]]; //容易错误
}
}
//cout<<i<<" "<<p<<" "<<imin<<endl;
qp[i]=p;
qv[i]=imin;
vist[i]=true;
}
};