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]
最小的人)。
示例 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
的观察在逻辑上是一致的原站题解
python3 解法, 执行用时: 72 ms, 内存消耗: 20.8 MB, 提交时间: 2022-11-29 11:29:37
class Solution: def loudAndRich(self, richer: List[List[int]], quiet: List[int]) -> List[int]: n = len(quiet) g = [[] for _ in range(n)] inDeg = [0] * n for r in richer: g[r[0]].append(r[1]) inDeg[r[1]] += 1 ans = list(range(n)) q = deque(i for i, deg in enumerate(inDeg) if deg == 0) while q: x = q.popleft() for y in g[x]: if quiet[ans[x]] < quiet[ans[y]]: ans[y] = ans[x] # 更新 x 的邻居的答案 inDeg[y] -= 1 if inDeg[y] == 0: q.append(y) return ans
golang 解法, 执行用时: 60 ms, 内存消耗: 7.6 MB, 提交时间: 2022-11-29 11:29:22
func loudAndRich(richer [][]int, quiet []int) []int { n := len(quiet) g := make([][]int, n) inDeg := make([]int, n) for _, r := range richer { g[r[0]] = append(g[r[0]], r[1]) inDeg[r[1]]++ } ans := make([]int, n) for i := range ans { ans[i] = i } q := make([]int, 0, n) for i, deg := range inDeg { if deg == 0 { q = append(q, i) } } for len(q) > 0 { x := q[0] q = q[1:] for _, y := range g[x] { if quiet[ans[x]] < quiet[ans[y]] { ans[y] = ans[x] // 更新 x 的邻居的答案 } inDeg[y]-- if inDeg[y] == 0 { q = append(q, y) } } } return ans }
javascript 解法, 执行用时: 92 ms, 内存消耗: 50.8 MB, 提交时间: 2022-11-29 11:29:07
/** * @param {number[][]} richer * @param {number[]} quiet * @return {number[]} */ var loudAndRich = function(richer, quiet) { const n = quiet.length; const g = new Array(n).fill(0); for (let i = 0; i < n; ++i) { g[i] = []; } const inDeg = new Array(n).fill(0); for (const r of richer) { g[r[0]].push(r[1]); ++inDeg[r[1]]; } const ans = new Array(n).fill(0); for (let i = 0; i < n; ++i) { ans[i] = i; } const q = []; for (let i = 0; i < n; ++i) { if (inDeg[i] === 0) { q.push(i); } } while (q.length) { const x = q.shift(); for (const y of g[x]) { if (quiet[ans[x]] < quiet[ans[y]]) { ans[y] = ans[x]; // 更新 x 的邻居的答案 } if (--inDeg[y] === 0) { q.push(y); } } } return ans; };
javascript 解法, 执行用时: 80 ms, 内存消耗: 51 MB, 提交时间: 2022-11-29 11:28:49
/** * @param {number[][]} richer * @param {number[]} quiet * @return {number[]} */ var loudAndRich = function(richer, quiet) { const n = quiet.length; const g = new Array(n).fill(0); for (let i = 0; i < n; ++i) { g[i] = []; } for (const r of richer) { g[r[1]].push(r[0]); } const ans = new Array(n).fill(-1); for (let i = 0; i < n; ++i) { dfs(i, quiet, g, ans); } return ans; }; const dfs = (x, quiet, g, ans) => { if (ans[x] !== -1) { return; } ans[x] = x; for (const y of g[x]) { dfs(y, quiet, g, ans); if (quiet[ans[y]] < quiet[ans[x]]) { ans[x] = ans[y]; } } }
golang 解法, 执行用时: 48 ms, 内存消耗: 7.8 MB, 提交时间: 2022-11-29 11:28:36
func loudAndRich(richer [][]int, quiet []int) []int { n := len(quiet) g := make([][]int, n) for _, r := range richer { g[r[1]] = append(g[r[1]], r[0]) } ans := make([]int, n) for i := range ans { ans[i] = -1 } var dfs func(int) dfs = func(x int) { if ans[x] != -1 { return } ans[x] = x for _, y := range g[x] { dfs(y) if quiet[ans[y]] < quiet[ans[x]] { ans[x] = ans[y] } } } for i := 0; i < n; i++ { dfs(i) } return ans }
python3 解法, 执行用时: 100 ms, 内存消耗: 20.8 MB, 提交时间: 2022-11-29 11:28:11
class Solution: def loudAndRich(self, richer: List[List[int]], quiet: List[int]) -> List[int]: n = len(quiet) g = [[] for _ in range(n)] for r in richer: g[r[1]].append(r[0]) ans = [-1] * n def dfs(x: int): if ans[x] != -1: return ans[x] = x for y in g[x]: dfs(y) if quiet[ans[y]] < quiet[ans[x]]: ans[x] = ans[y] for i in range(n): dfs(i) return ans