题目:
一场比赛中共有 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
提交通过: