题目:
一场比赛中共有 n 支队伍,按从 0 到 n - 1 编号。每支队伍也是 有向无环图(DAG) 上的一个节点。
给你一个整数 n 和一个下标从 0 开始、长度为 m 的二维整数数组 edges 表示这个有向无环图,其中 edges[i] = [ui, vi] 表示图中存在一条从 ui 队到 vi 队的有向边。
从 a 队到 b 队的有向边意味着 a 队比 b 队 强 ,也就是 b 队比 a 队 弱 。
在这场比赛中,如果不存在某支强于 a 队的队伍,则认为 a 队将会是 冠军 。
如果这场比赛存在 唯一 一个冠军,则返回将会成为冠军的队伍。否则,返回 -1 。
注意
- 环 是形如
a1, a2, ..., an, an+1的一个序列,且满足:节点a1与节点an+1是同一个节点;节点a1, a2, ..., an互不相同;对于范围[1, n]中的每个i,均存在一条从节点ai到节点ai+1的有向边。 - 有向无环图 是不存在任何环的有向图。
思考:
1. 因为使用有向无环图记录队伍的强弱关系,所以如果a队是冠军,那么一定没有指向a的边。同理,如果有指向a的边,那么a一定不是冠军。遍历edges数组,用字典记录指向每个队伍的边的个数(入度),最后字典中值为0的队伍有可能是冠军。
2. “冠军是唯一的”,所以只要字典中有一个以上的队伍值为0,返回-1
代码如下:
class Solution(object):
def findChampion(self, n, edges):
"""
:type n: int
:type edges: List[List[int]]
:rtype: int
"""
count_dict = {key:0 for key in range(n)}
# 字典记录每个队的比自己强的队的数量(每个节点的入度)
win = 0
for u, v in edges:
count_dict[v] += 1
for key, value in count_dict.items():
if value == 0:
win += 1
if win == 1:
winner = key
else:
return -1
return winner
提交通过:
寻找有向无环图中唯一冠军队伍:基于入度的算法,
文章描述了一种算法,通过计算有向无环图中各队伍的入度来确定是否存在唯一冠军队伍。如果有且仅有一个队伍的入度为0,则该队伍是冠军;否则返回-1。
1429

被折叠的 条评论
为什么被折叠?



