BFS,题意比较难懂,是求离query L层的总共人数
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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114 |
#include <stdio.h> #include <string.h> #include <iostream> #include <algorithm> #include <map> #include <vector> #include <queue> using
namespace std; const
int N = 1005; struct
E { int
node; }; vector<E> v[N]; int
mark[N]; void
add_edge( int
a, int
b) { E tmp; tmp.node = b; v[a].push_back(tmp); } struct
ANS { int
num; int
lev; }; void
BFS( int
x, int
ll) { queue<ANS> q; ANS tmp; tmp.lev = 0; tmp.num = x; q.push(tmp); mark[x] = 1; int
level = 0; int
ans = 0; while
(!q.empty()) { ANS top = q.front(); int
node = top.num; int
lev = top.lev; if
(lev >= ll) break ; q.pop(); for
( int
i = 0; i < v[node].size(); i++) { if
(mark[v[node][i].node] == 0) { mark[v[node][i].node] = 1; tmp.lev = lev + 1; tmp.num = v[node][i].node; //printf("%d %d\n", tmp.num, tmp.lev); ans++; q.push(tmp); } } } printf( "%d\n" , ans); } int
main() { int
n, ll; while
(scanf( "%d%d" , &n, &ll) != EOF) { for
( int
i = 1; i <= n; i++) { int
k, a; scanf( "%d" , &k); while
(k--) { scanf( "%d" , &a); add_edge(a, i); } } int
query_num, query; scanf( "%d" , &query_num); while
(query_num--) { scanf( "%d" , &query); memset(mark, 0, sizeof (mark)); BFS(query, ll); } } return
0; } |
Forwards on Weibo (30),布布扣,bubuko.com
原文:http://www.cnblogs.com/echobfy/p/3576803.html