题目:
有一组 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] 最小的人)。
解答:
class Solution:
def loudAndRich(self, richer: List[List[int]], quiet: List[int]) -> List[int]:
n=len(quiet)
g=[[] for _ in range(n)]
for a,b in richer:
g[b].append(a)
#print(dic)
res=[-1]*n
def dfs(x):
if res[x]!=-1:
return
#若没有人 拥有的钱比他多,则返回自身
res[x]=x
for y in g[x]:
dfs(y)
#若更安静,则替换
if quiet[res[y]]<quiet[res[x]]:
res[x]=res[y]
for i in range(n):
dfs(i)
return res