Newcoder 38 D.珂朵莉的无向图（bfs）

Description

珂朵莉给了你一个无向图，每次查询给$t$个点以及一个常数$s$，求有多少个图中的点距离给出的那$t$个点中至少一个距离$\le s$

Input

第一行三个数表示$n，m，q$

之后$m$行每行两个数$u，v$表示有一条边位于$u$和$v$两个点之间

之后 $2q$ 行表示询问

每次询问先输入两个数$t，s$

之后一行$t$个数，表示$t$个特殊点

$(1\le n,m,q\le 5000,\sum t\le 5\cdot 10^5,s\le 10^9)$

Output

$q$行，每行一个数表示答案

Sample Input

5 6 6
2 3
1 3
2 5
1 3
3 2
2 5
1 1
3
1 1
1
1 4
1
1 2
5
1 4
1
1 4
5

Sample Output

3
2
4
3
4
4

Solution

每次以所给$t$个点为起点$bfs$整张图即可，时间复杂度$O((n+m)q)$

Code

#include<cstdio>
#include<vector>
#include<queue>
using namespace std;
const int maxn=5005;
int n,m,q,dis[maxn],vis[maxn];
vector<int>g[maxn];
int main()
{
	scanf("%d%d%d",&n,&m,&q);
	while(m--)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		g[u].push_back(v),g[v].push_back(u);
	}
	while(q--)
	{
		int t,s;
		scanf("%d%d",&t,&s);
		for(int i=1;i<=n;i++)dis[i]=s+1,vis[i]=0;
		queue<int>que;
		for(int i=1;i<=t;i++)
		{
			int u;
			scanf("%d",&u);
			que.push(u);
			dis[u]=0;
			vis[u]=1;
		}
		while(!que.empty())
		{
			int u=que.front();
			que.pop();
			if(dis[u]==s)break;
			for(int i=0;i<g[u].size();i++)
			{
				int v=g[u][i];
				if(vis[v])continue;
				vis[v]=1,dis[v]=dis[u]+1,que.push(v);
			}
		}
		int res=0;
		for(int i=1;i<=n;i++)
			if(dis[i]>s)res++;
		printf("%d\n",n-res);
	}
	return 0;
}