题意:无向图的每个顶点有一个属性wi,Q个询问,第i个询问给定顶点p和价值a,问距离p最近的wi<a的i距离p有多远
题解:
w的范围很小,直接做100次BFS,j计算d[i][j]表示离i最近的属性值恰号为j的点的距离即可。时间复杂度O(100(n+m)+Q)。
代码:
#include<iostream>
#include<queue>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=110,M=100010;
int h[M],e[M*2],ne[M*2],idx;
int n,m,q;
int w[M];
int d[M][N]; //d[i][j]距离i最近的属性值恰号为j的点的距离
bool st[M];
void add(int a,int b) //建边
{
e[idx]=b;ne[idx]=h[a];h[a]=idx++;
}
void bfs(int s)
{
queue<int>qq;
memset(st,false,sizeof st);
for(int i=1;i<=n;i++)
if(w[i]==s)
{
qq.push(i);
st[i]=true;
d[i][s]=0;
}
while(qq.size())
{
int t=qq.front();
qq.pop();
for(int i=h[t];i!=-1;i=ne[i])
{
int j=e[i];
if(st[j]) continue; //一层一层的遍历,往回遍历就跳过
d[j][s]=d[t][s]+1;
qq.push(j);
st[j]=true;
}
}
}
int main()
{
scanf("%d%d%d",&n,&m,&q);
for(int i=1;i<=n;i++) scanf("%d",&w[i]); //输入每个点属性值
memset(h,-1,sizeof h); //链表初始化
int a,b;
for(int i=0;i<m;i++)
{
scanf("%d%d",&a,&b);
if(a==b) continue;
add(a,b);add(b,a); //建立无向边
}
memset(d,-1,sizeof d); //初始化距离为-1
for(int i=1;i<=100;i++) //按照质量宽搜
bfs(i);
while(q--)
{
int p,a;
scanf("%d%d",&p,&a);
int minn=0x3f3f3f3f;
for(int i=1;i<=a;i++)
if(d[p][i]!=-1)
minn=min(minn,d[p][i]); //找到小于忍耐值a中的最短距离
if(minn==0x3f3f3f3f) minn=-1;
printf("%d\n",minn);
}
return 0;
}