题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/loud-and-rich
【题目描述】
有一组 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] 最小的人)。
【示例】
示例 1:
输入: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。
其他的答案也可以用类似的推理来解释。
示例 2:
输入:richer = [], quiet = [0]
输出:[0]
提示:
- n == quiet.length
- 1 <= n <= 500
- 0 <= quiet[i] < n
- quiet 的所有值互不相同
- 0 <= richer.length <= n * (n - 1) / 2
- 0 <= ai, bi < n
- ai != bi
- richer 中的所有数对 互不相同
- 对 richer 的观察在逻辑上是一致的
【解题思路】
根据题意,richer[i]=[ai, bi]表示person ai比person bi更有钱。如果此时ai的安静值小于bi的安静值,则把bi的安静值更新为ai的安静值,表示当前情况下,对于bi来说ai的安静值为当前最优安静值(用列表q表示)。
重复上述过程,直到所有人的最优安静值均已确定,即q不再变化。
因为每个人的安静值都不相同,所以可以从数组quiet中确定每个人的最优安静值所对应的人的编号,该编号即为最终输出。
【代码】
class Solution:
def loudAndRich(self, richer: List[List[int]], quiet: List[int]) -> List[int]:
q,l = [], len(quiet) # q:最优及当前最优的安静值 l:人数
q.extend(quiet) # 复制quiet到q,以便后续比较
min_p = [0] * l # 最终输出
f = True # q是否更新
# 迭代求解最优安静值
while(f):
f = False
for i,r in enumerate(richer):
if q[r[0]]<q[r[1]]:
q[r[1]] = q[r[0]]
f = True
# 根据最优安静值确定输出结果
for i in range(l):
min_p[i] = quiet.index(q[i])
return min_p
【提交结果】