The 2021 Sichuan Provincial 四川省赛 L. Spicy Restaurant(多源bfs)

LINK

由于$w_i<=100$

所以直接枚举一个权值$k$,把所有$w_i=k$的点作为源点做一次多源$bfs$

那么可以预处理$dis[i][j]$表示$j$点距离最近的辣味$i$多远

然后顺着更新一遍$\mathrm{d}\mathrm{i}\mathrm{s}[\mathrm{i}][\mathrm{j}]=\mathrm{m}\mathrm{i}\mathrm{n}(\mathrm{d}\mathrm{i}\mathrm{s}[\mathrm{i}-1][\mathrm{j}],\mathrm{d}\mathrm{i}\mathrm{s}[\mathrm{i}][\mathrm{j}])$

#include <bits/stdc++.h>
using namespace std;
const int maxn = 2e5+10;
int n,m,q,inf,dis[109][maxn],w[maxn];
vector<int>vec[maxn];
void BFS(int c)
{
	queue<int>q;
	for(int j=1;j<=n;j++)	
		if( w[j]==c )	q.push( j ),dis[c][j] = 0;	
	while( !q.empty() )
	{
		int u = q.front(); q.pop();
		for(auto v:vec[u] )
		{
			if( dis[c][v]!=inf )	continue;
			q.push( v ); dis[c][v] = dis[c][u]+1;
		}
	}
}
int main()
{
	memset( dis,0x3f,sizeof dis );
	inf = dis[0][0];
	cin >> n >> m >> q;
	for(int i=1;i<=n;i++)	scanf("%d",&w[i] );
	for(int i=1;i<=m;i++)
	{
		int l,r;scanf("%d%d",&l,&r);
		vec[l].push_back( r ); vec[r].push_back( l );
	}
	for(int i=100;i>=1;i--)	BFS( i );
	for(int i=1;i<=n;i++)
	for(int j=1;j<=100;j++)
		dis[j][i] = min( dis[j][i],dis[j-1][i] );
	while( q-- )
	{
		int l,r; scanf("%d%d",&r,&l);
		printf("%d\n",dis[l][r]==inf?-1:dis[l][r] );
	}
}