Leetcode 851. 喧闹和富有 C++

Leetcode 851. 喧闹和富有

题目

在一组 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. 1 <= quiet.length = N <= 500
  2. 0 <= quiet[i] < N,所有 quiet[i] 都不相同。
  3. 0<= richer.length <= N * (N-1) / 2
  4. 0 <= richer[i][j] < N
  5. richer[i][0]!= richer[i][1]
  6. richer[i] 都是不同的。
  7. 对 richer 的观察在逻辑上是一致的。

题解

bfs拓扑排序的思想
ans[x] = y存储所有拥有的钱不少于 person x 的人中,person y 是最安静的人。
我们依次获取较富有的人,找到比它不富有的人中最安静的人,假设,我们存在richer[] = [x,y],那么考虑ans[x]时,若满足quiet[ans[y]] > quiet[ans[x]],则ans[x] = ans[y],因为ans[y]比y要富有,且最安静,而x比y要富有,满足式子时,ans[y]就最安静。
详细过程见代码

代码

	vector<int> loudAndRich(vector<vector<int>>& richer, vector<int>& quiet) {
        unordered_map<int,vector<int>> graph;
        int n = richer.size(),num = quiet.size();
        vector<int> deg(num,0);
        for(int i=0; i<n; i++){
            graph[richer[i][0]].push_back(richer[i][1]);
            deg[richer[i][1]]++;		//记录入度
        }
        vector<int> ans(num,0);
        vector<int> visit(num,0);
        for(int i=0; i<num; i++)
            ans[i] = i;
        queue<int> q;
        for(int i=0; i<num; i++){		//从最富有的人开始遍历,也就是入度为0的人
            if(deg[i] == 0){
                q.push(i);
            }
        }
        int size;
        while(!q.empty()){
            size = q.size();
            while(size--){
                int now = q.front();
                visit[now] = 1;
                q.pop();
                for(int i=0; i<graph[now].size(); i++){		//删除当前节点及对应边,也就是删除相连节点的入度
                    deg[graph[now][i]]--;
                    if(quiet[ans[graph[now][i]]] > quiet[ans[now]]){
                        ans[graph[now][i]] = ans[now];
                    }
                }
            }
            for(int i=0; i<num; i++){		//从最富有的人开始遍历,也就是入度为0的人
                if(!visit[i] && deg[i]==0){
                    q.push(i);
                }
            }
        }
        return ans;
    }

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/loud-and-rich
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

你可能感兴趣的:(Leetcode 851. 喧闹和富有 C++)