牛客练习赛14 E - 无向图中的最短距离 (bfs+bitset)

牛客练习赛14 E - 无向图中的最短距离 (bfs+bitset)

链接

https://ac.nowcoder.com/acm/contest/82/E

题目描述
有一个n个点的无向图，有m次查询，每次查询给出一些(xi,yi)

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

每次查询图中有多少个点v与至少一个这次询问给出的(xi,yi)满足dist(v,xi)<=yi

输入描述:
第一行三个数表示n，m，q

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

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

之后一行2a个数，第2i-1个数xi和第2i个数yi表示一个二元组(xi,yi)
输出描述:
输出q行，每行一个数表示这次询问的答案
示例1
输入
复制
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
输出
复制
3
2
4
3
4
3
备注:
对于100%的数据，n <= 1000 , m <= 100000 , q <=
100000
a的和<= 2100000

思路：
首先对于每一个节点进行bfs找到它对其他节点的最短路。
用数组 d[i][j] 来维护 i到j的距离，
那么我们定理一个 bitset<1011> bit[1011][1011]
bit[i][j][k] 代表 第i个节点距离小于等于j的有没有k，有没有只有两个状态，即0/1
所以用　ｂｉｔｓｅｔ可以完美维护。
对于处理，我们对每一个节点ｂｆｓ后，更新 b
又因为是 小于等于， 所以 bit[i][j][k] =1的话， 那么bit[i][j~n][k] 都为1.
这样我们就可以从1到n,用bitset 的或运算|=来更新使距离j更大的一定包含更小的情况。
然后对于询问，
我们只需要让一个全是0的bitset 去|= 那num个 b[x][y] 即可。
最后得到的bitset中的1的个数就是答案数。

代码 ：

#include<bits/stdc++.h>
using namespace std ;
const int maxn = 1010 ;
bitset<maxn>bit[maxn][maxn] ;
//bit[i][j][k] 代表 第i个节点距离小于等于j的有没有k
int d[maxn][maxn] , n ;
bool is[maxn] ;
vector<int> p[maxn] ;
queue<int> q ;
void bfs(int id){			//求到id点的最短路径 
	memset(is , 0 , sizeof(is)) ;
	is[id] = 1 ;
	d[id][id] = 0 ;
	q.push(id) ;
	while(!q.empty()){
		int v = q.front() ;
		 q.pop() ;
		 int len = p[v].size() ;
		 for(int i = 0 ; i < len ; i++){
		 	int to = p[v][i] ;
		 	if(!is[to]){
		 		is[to] = 1 ;
		 		q.push(to) ;
		 		d[id][to] = d[id][v] + 1 ;
			 }
		 }
	}
}

int main(){
	int m , que ;
	cin >> n >> m >> que ;
	for(int i = 0 ; i < m ; i++){
		int x , y ;
		cin >> x >> y ;
		p[x].push_back(y) ;
		p[y].push_back(x) ;
	}
	memset(d , -1 , sizeof(d)) ;
	for(int i = 1 ; i <= n ; i++){
		bfs(i) ;
		for(int j = 0 ; j <= n ; j++)
			if(d[i][j] != -1)	bit[i][d[i][j]][j] = 1 ;
		//存在路径 更新bit 
		
		for(int j = 1 ; j <= n ; j++)
			bit[i][j] |= bit[i][j-1] ;
		//更新长于j的长度 
	}
	while(que--){
		int t ;
		cin >> t ;
		bitset<maxn>ans ;
		while(t--){
			int x , y ;
			cin >> x >> y ;
			ans |= bit[x][y] ;
		}
		cout << ans.count() << endl ;
	}
	return 0 ;
}