LightOJ 1111 Best Picnic Ever 简单bfs

题目：http://www.lightoj.com/volume_showproblem.php?problem=1111

题意：有n个城市k个人，城市间有m条单向路，现在k个人想到集中到某一个城市，问这样的城市有多少个

思路：分别从k个人所在的城市进行一个bfs，记录每个城市被访问的次数，把访问到的城市均加1，最后被访问次数等于k的城市就是满足条件的

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <stack>
#include <string>
#include <queue>
using namespace std;

const int N = 1010;
struct edge
{
int to, next;
}g[N*10];
bool vis
;
int cnt, head
;
int arr
, num
;
int cas;
void add_edge(int v, int u)
{
g[cnt].to = u, g[cnt].next = head[v], head[v] = cnt++;
}
void bfs(int s)
{
memset(vis, 0, sizeof vis);
queue<int> que;
que.push(s), vis[s] = true, num[s]++;
while(! que.empty())
{
int v = que.front(); que.pop();
for(int i = head[v]; i != -1; i = g[i].next)
{
int u = g[i].to;
if(! vis[u])
que.push(u), vis[u] = true, num[u]++;
}
}
}
int main()
{
int t, k, n, m, a, b;
scanf("%d", &t);
while(t--)
{
cnt = 0;
memset(head, -1, sizeof head);
memset(num, 0, sizeof num);
scanf("%d%d%d", &k, &n, &m);
for(int i = 1; i <= k; i++) scanf("%d", &arr[i]);
for(int i = 1; i <= m; i++) scanf("%d%d", &a, &b), add_edge(a, b);
for(int i = 1; i <= k; i++) bfs(arr[i]);
int res = 0;
for(int i = 1; i <= n; i++) if(num[i] == k) res++;
printf("Case %d: %d\n", ++cas, res);
}
return 0;
}