leetcode：851. 喧闹和富有

题目来源

• leetcode：851. 喧闹和富有

题目描述

class Solution {
public:
    vector<int> loudAndRich(vector<vector<int>>& richer, vector<int>& quiet) {

    }
};

题目解析

分析

题目意思就是要找出 已知比某个人「富有」的人中最安静的人，可以分两个部分去考虑：

1. 找出所有比其他人富有的人
2. 找出其中最安静的人

由于题目中只给出了部分偏序的信息，没有绝对的值。所以我们不能确定每个人之间的富有程度；但是基于a>b & b>c => a>c 这样的传递关系；我们是可以找出一部分确定比我们富有的人的。

具体做法就是要对整个富有关系建图；图中的节点表示人，图中的边是有向边，代表两人之间富有程度的比较关系，边由更贫穷的一方指向更又富有的一方

• 对于第一个问题：我们从图中的任意一个节点触发，进行深度优先遍历，所有的后继节点都是比自己富有的
• 对于第二个问题：我们在遍历的时候维护一下最穷的人是谁即可

举个例子

对于例子1，可以建立如下图：

算法

拓扑排序

• 对于入度为0的人有：6、4、1、2，它们都填写自己，因为没有人比它有钱
• 对于入队为1的人有：
  • 3：比它有钱的人有：3、4、5、6，其中最安静的是5

我的答案

class Solution {
    struct Info{
        int person;
        int quiet;
    };
public:
    vector<int> loudAndRich(vector<vector<int>>& richer, vector<int>& quiet) {
        int N = quiet.size();
        std::vector<int> inDegree(N);
        std::map<int, std::set<int>> map;
        for(auto iter : richer){
            map[iter[0]].insert(iter[1]);
            ++inDegree[iter[1]];
        }

        std::vector<Info> tmp(N);
        std::queue<int> queue;
        for (int i = 0; i < N; ++i) {
            if(inDegree[i] == 0){
                queue.push(i);
            }
            tmp[i].person = i;
            tmp[i].quiet = quiet[i];
        }

        while (!queue.empty()){
            auto top = queue.front(); queue.pop();
            for(auto next : map[top]){
                if(tmp[top].quiet < tmp[next].quiet){    // top一定比next有钱，那就看是不是更加低调了(它具有传递性，所以用tmp来考虑)
                    tmp[next].quiet = tmp[top].quiet;
                    tmp[next].person = tmp[top].person;
                }
                inDegree[next]--;
                if(inDegree[next] == 0){
                    queue.push(next);
                }
            }
        }


        std::vector<int> ans(N);
        for(int i = 0; i < tmp.size(); i++){
            ans[i] = tmp[i].person;
        }

        return ans;
    }
};