题目
有一组 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] 最小的人)。
解题思路
记忆化搜索。利用贫富关系反向建图,让穷的去找富的。然后用DFS对所有节点搜索比当前节点富的节点,当某个节点所有子节点都搜索到,就找他们的最小值,并且更新ans[nowNode]。关于剪枝,如果在开始搜索的时候或者搜索的过程中遇到已经有值的节点,直接拿值即可。
在更新节点值的时候,懒得直接用下标作为最终值了,又quiet的值唯一,所以构建了个map映射了他们的关系,这里写好的话可以省一点内存。
代码
class Solution {
private int[] map;
private int[] ans;
private int[] quiet;
private int lengthQ;
private List<Integer>[] pic;
public int[] loudAndRich(int[][] richer, int[] quiet) {
lengthQ = quiet.length;
this.quiet = quiet;
map = new int[lengthQ];
ans = new int[lengthQ];
pic = new List[lengthQ];
for (int i = 0; i < lengthQ; i++) {
pic[i] = new ArrayList<>();
map[quiet[i]] = i;
ans[i] = -1;
}
for (int[] ints : richer) {
int poor = ints[1], rich = ints[0];
pic[poor].add(rich);
}
for (int i = 0; i < lengthQ; i++) if (ans[i] == -1) dfs(i);
for (int i = 0; i < lengthQ; i++) ans[i] = map[ans[i]];
return ans;
}
private int dfs(int nowNode) {
int minQuiet = quiet[nowNode];
if (pic[nowNode].size() > 0) {
for (Integer nextNode : pic[nowNode]) {
if (ans[nextNode] != -1) minQuiet = Math.min(minQuiet, ans[nextNode]);
else minQuiet = Math.min(minQuiet, dfs(nextNode));
}
}
ans[nowNode] = minQuiet;
return minQuiet;
}
}