851. 喧闹和富有

题目

有一组 n 个人作为实验对象，从 0 到 n - 1 编号，其中每个人都有不同数目的钱，以及不同程度的安静值（quietness）。为了方便起见，我们将编号为 x 的人简称为 "person x "。

给你一个数组 richer ，其中 richer[i] = [ai, bi] 表示 person ai 比 person bi 更有钱。另给你一个整数数组 quiet ，其中 quiet[i] 是 person i 的安静值。richer 中所给出的数据 逻辑自恰（也就是说，在 person x 比 person y 更有钱的同时，不会出现 person y 比 person x 更有钱的情况 ）。

现在，返回一个整数数组 answer 作为答案，其中 answer[x] = y 的前提是，在所有拥有的钱肯定不少于 person x 的人中，person y 是最安静的人（也就是安静值 quiet[y] 最小的人）。

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

代码

class Solution {
     int[] ans ;//储存答案
     ArrayList<ArrayList<Integer>> moreRicher;//储存比当前结点更富有的人
    public int[] loudAndRich(int[][] richer, int[] quiet) {
        int n= quiet.length;
        moreRicher = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            moreRicher.add(new ArrayList<>());
        }
        for (int i = 0; i <richer.length ; i++) {
            int  p1=richer[i][0];//p1比p2更富有
            int  p2=richer[i][1];
            moreRicher.get(p2).add(p1);//记录
        }
        ans =new int[n];
        for (int i = 0; i <n ; i++) {
            ans[i]=-1;
        }
        int i =0;
        for (int j = 0; j <n ; j++) {
            getricher(moreRicher,j,quiet);
        }
        return ans;
    }

    void getricher(ArrayList<ArrayList<Integer>> a, int x,int[] quiet){//记录第x个人的最安静值
        if (ans[x]!=-1) {//因为遍历比较穷的人的时候，可能已经更新过较富的人的答案了，所以可能已经更新过了
            return;
        }
        ans[x]=x;//若更新到新结点，则不需要后续再遍历了，至少赋值是自己
        for (int y:a.get(x)
             ) {
            getricher(moreRicher,y,quiet);//深度优先遍历
            if (quiet[ans[y]]<quiet[ans[x]]) ans[x]=ans[y];//如果比他富的安静值小，则更新

        }
    }
}

要点

• 本题需要利用图结构，记录有向无环图，然后DFS深度优先遍历
• 因为DFS的时候可能已经更新了后续的结点(传导性)，所以需要判断是否已经更新后续节点，减少复杂度，否则会超时