Newcoder 82 E.无向图中的最短距离（bfs+bitset）

Description

有一个$n$个点的无向图，有$m$次查询，每次查询给出一些$(x_i,y_i)$

令$dist(x,y)$表示$x$和$y$点在图中最短距离，$dist(x,x)=0$,如果$x,y$不连通则$dist(x,y)=inf$

每次查询图中有多少个点$v$与至少一个这次询问给出的$(x_i,y_i)$满足$dist(v,x_i)\le y_i$

Input

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

之后$m$行每行两个数$x,y$表示有一条$x$与$y$之间的边，边权为$1$

之后$q$次询问，每个询问先给你一个数$a$

之后一行$2a$个数，第$2i-1$个数$x_i$和第$2i$个数$y_i$表示一个二元组$(x_i,y_i)$

$(n\le 10^3,m\le 10^5,q\le 10^5,\sum |a|\le 2100000)$

Output

输出$q$行，每行一个数表示这次询问的答案

Sample Input

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

Sample Output

3
2
4
3
4
3

Solution

从每点开始$bfs$，用$f[i][j]$这个$bitset$记录所有与$i$点距离不超过$j$的点，那么对于一组查询，把所有$f[x_i][y_i]$或一下统计$1$的个数即可，时间复杂度$O(\frac{n^3}{32})$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
#include<bitset>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=1005;
bitset<1005>f[maxn][maxn],ans;
int n,m,q,dis[maxn];
vector<int>g[maxn];
void bfs(int s)
{
	queue<int>que;
	que.push(s);
	f[s][0].set(s);
	memset(dis,-1,sizeof(dis));
	dis[s]=0;
	while(!que.empty())
	{
		int u=que.front();que.pop();
		for(int i=0;i<g[u].size();i++)
		{
			int v=g[u][i];
			if(dis[v]!=-1)continue;
			dis[v]=dis[u]+1;
			que.push(v);
			f[s][dis[v]].set(v);
		}
	}
	for(int i=1;i<=n;i++)f[s][i]|=f[s][i-1];
}
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);
	}
	for(int i=1;i<=n;i++)bfs(i);
	while(q--)
	{
		int a,x,y;
		scanf("%d",&a);
		ans.reset();
		while(a--)
		{
			scanf("%d%d",&x,&y);
			y=min(y,n);
			ans|=f[x][y];
		}
		printf("%d\n",ans.count());
	}
	return 0;
}