【力扣每日一题——2374. 边积分最高的节点】python

2374. 边积分最高的节点

给你一个有向图，图中有 n 个节点，节点编号从 0 到 n - 1 ，其中每个节点都 恰有一条 出边。

图由一个下标从 0 开始、长度为 n 的整数数组 edges 表示，其中 edges[i] 表示存在一条从节点 i 到节点 edges[i] 的 有向 边。

节点 i 的 边积分 定义为：所有存在一条指向节点 i 的边的节点的 编号 总和。

返回 边积分 最高的节点。如果多个节点的 边积分 相同，返回编号 最小 的那个。

示例 1：

输入：edges = [1,0,0,0,0,7,7,5]
输出：7
解释：
- 节点 1、2、3 和 4 都有指向节点 0 的边，节点 0 的边积分等于 1 + 2 + 3 + 4 = 10 。
- 节点 0 有一条指向节点 1 的边，节点 1 的边积分等于 0 。
- 节点 7 有一条指向节点 5 的边，节点 5 的边积分等于 7 。
- 节点 5 和 6 都有指向节点 7 的边，节点 7 的边积分等于 5 + 6 = 11 。
节点 7 的边积分最高，所以返回 7 。

示例 2：

输入：edges = [2,0,0,2]
输出：0
解释：
- 节点 1 和 2 都有指向节点 0 的边，节点 0 的边积分等于 1 + 2 = 3 。
- 节点 0 和 3 都有指向节点 2 的边，节点 2 的边积分等于 0 + 3 = 3 。
节点 0 和 2 的边积分都是 3 。由于节点 0 的编号更小，返回 0 。

提示：

• n == edges.length
• 2 <= n <= 10**5
• 0 <= edges[i] < n
• edges[i] != i

# 一眼图论！好，错了
# 认真读题并且看过样例之后就会发现是个统计的题，代码如下
class Solution:
    def edgeScore(self, edges: List[int]) -> int:
        n = len(edges)
        d = {i : 0 for i in range(n)}
        for i,j in enumerate(edges):
            d[j] = d[j] + i
        d = list(sorted(d.items(), key = lambda items: (items[1],items[0])))
        for i,j in d:
            if j == d[-1][1]:
                return i
# 较为复杂了，不应该用字典来存的，以下是改正之后的（照着灵神改的，tql）
class Solution:
    def edgeScore(self, edges: List[int]) -> int:
        n = len(edges)
        score = [0] * n
        for i, to in enumerate(edges):
            score[to] += i
        return score.index(max(score))