forked from zzjlogin/ccf_certification
-
Notifications
You must be signed in to change notification settings - Fork 0
/
201709-4.py
96 lines (80 loc) · 3.05 KB
/
201709-4.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
#!/usr/bin/env python
# -*- coding: utf-8 -*-
'''
问题描述
某国的军队由N个部门组成,为了提高安全性,部门之间建立了M条通路,每条通路只能单向传递信息,即一条从部门a到部门b的通路只能由a向b传递信息。信息可以通过中转的方式进行传递,即如果a能将信息传递到b,b又能将信息传递到c,则a能将信息传递到c。一条信息可能通过多次中转最终到达目的地。
由于保密工作做得很好,并不是所有部门之间都互相知道彼此的存在。只有当两个部门之间可以直接或间接传递信息时,他们才彼此知道对方的存在。部门之间不会把自己知道哪些部门告诉其他部门。
上图中给了一个4个部门的例子,图中的单向边表示通路。部门1可以将消息发送给所有部门,部门4可以接收所有部门的消息,所以部门1和部门4知道所有其他部门的存在。部门2和部门3之间没有任何方式可以发送消息,所以部门2和部门3互相不知道彼此的存在。
现在请问,有多少个部门知道所有N个部门的存在。或者说,有多少个部门所知道的部门数量(包括自己)正好是N。
输入格式
输入的第一行包含两个整数N, M,分别表示部门的数量和单向通路的数量。所有部门从1到N标号。
接下来M行,每行两个整数a, b,表示部门a到部门b有一条单向通路。
输出格式
输出一行,包含一个整数,表示答案。
样例输入
4 4
1 2
1 3
2 4
3 4
样例输出
2
样例说明
部门1和部门4知道所有其他部门的存在。
评测用例规模与约定
对于30%的评测用例,1 ≤ N ≤ 10,1 ≤ M ≤ 20;
对于60%的评测用例,1 ≤ N ≤ 100,1 ≤ M ≤ 1000;
对于100%的评测用例,1 ≤ N ≤ 1000,1 ≤ M ≤ 10000。
'''
#以下提交是25分:
d_start = {}
d_end = {}
c = 0
a = list(map(int, input().split()))
n, m = a[0], a[1]
s = set()
def graph(e):
if e[0] in d_start:
d_start.get(e[0]).append(e[1])
else:
d_start[e[0]] = [e[1]]
if e[1] not in d_start:
d_start[e[1]] = []
def graph2(e):
if e[0] in d_end:
d_end.get(e[0]).append(e[1])
else:
d_end[e[0]] = [e[1]]
if e[1] not in d_end:
d_end[e[1]] = []
for i in range(m):
j = list(map(int, input().split()))
for k in j:
s.add(k)
graph(j)
j.reverse()
graph2(j)
def dfs_traverse(graph, start):
visited, stack = set(), [start]
while stack:
node = stack.pop()
if node not in visited:
visited.add(node)
if node in graph:
for nextNode in graph.get(node):
if nextNode not in visited:
stack.append(nextNode)
return visited
for i in d_start:
p = dfs_traverse(d_start, i)
if set(p) == s:
c += 1
else:
pass
for i in d_end:
p = dfs_traverse(d_end, i)
if set(p) == s:
c += 1
else:
pass
print(c)