L. Spicy Restaurant（多源BFS+递推）

L. Spicy Restaurant

题目链接

https://codeforces.com/gym/103117/problem/L

思路：

1、考虑到1≤ai≤100 ，一开始直接使用BFS，对每个点进行100次BFS，得到d[i][j]，表示最近的点到j点满足辣味<=i的距离，这样的时间复杂度为O(100n^2+q)，q为询问个数，会超时。
2、用多源BFS，得到dis[i][j],表示j点与最近的辣味等于i的点之间的距离，递推处理再更新一遍，得到dis[i][j]表示j点与最近的辣味<=i的点之间的距离，这样BFS是O(100（n+m）),递推更新是O（100n）,合起来应该是O(100（n+m）+q)。

细节

1、关闭流同步ios::sync_with_stdio(false);
2、BFS时对已经访问过的点先标记，再放入队列中，否则会造成冗余（调了超久，提交的时候一直是WA test2,想不明白哎，之前也这么干过一次）

AC代码

#include <bits/stdc++.h>
using namespace std;

vector<int> nn[100010];
int w[100010];

bool vis[100010];
int d[105][100010];

int n,m,q; 

//使用二维数组+队列替换队列+结构体 

//进行100次多源BFS  
void BFS(int tt){
	
	queue<int>q; 
	memset(vis,false,sizeof vis );
	
	for(int i=1;i<=n;i++){
		if(w[i]==tt){
			q.push(i);
			vis[i]=true;
			d[tt][i]=0;
		}
	}
	
	while(!q.empty()){
		int cur=q.front();
		q.pop();
//		vis[cur]=true; 	//取出才标记访问会导致错误?冗余 
		for(auto a:nn[cur]){
			if(vis[a]) continue;
			vis[a]=true;
			d[tt][a]=d[tt][cur]+1;
			q.push(a);
		} 
	}
} 

int main() {
	ios::sync_with_stdio(false);
	 
	memset(d,-1,sizeof d );
	
	//q为tourists的数量，即询问 
	cin>>n>>m>>q;
	
	for(int i=1;i<=n;i++){
		cin>>w[i];
	}
	
	for(int i=0;i<m;i++){
		int a,b;
		cin>>a>>b;
		
		nn[a].push_back(b);
		nn[b].push_back(a);
		
	}
	
	//1、多源BFS得到dis[i][j],表示j点最近的辣味等于i的点之间的距离
//	for(int i=1;i<=100;i++) BFS(i);
	for(int i=100;i>=1;i--)	BFS(i);
	//2、再更新一遍，得到dis[i][j]表示j点最近的辣味<=i之间的距离
	for(int i=1;i<=n;i++)
		for(int j=2;j<=100;j++)
			if(d[j][i]!=-1 && d[j-1][i]!=-1) d[j][i]=min(d[j][i],d[j-1][i]); 
			else if(d[j][i]==-1) d[j][i]=d[j-1][i]; 

	//query
	int o,oo;
	for(int i=0;i<q;i++){
		cin>>o>>oo;
		cout<<d[oo][o]<<endl;	
	} 
	 
	return 0;
	
}

参考：

https://blog.csdn.net/jziwjxjd/article/details/117750292