L. Spicy Restaurant

L. Spicy Restaurant
L. Spicy Restaurant_第1张图片
L. Spicy Restaurant_第2张图片

题意:无向图的每个顶点有一个属性wi,Q个询问,第i个询问给定顶点p和价值a,问距离p最近的wi 题解:
w的范围很小,直接做100次BFS,j计算d[i][j]表示离i最近的属性值恰号为j的点的距离即可。时间复杂度O(100(n+m)+Q)。
在这里插入图片描述

代码:

#include 
#include
#include
#include 

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;
}

你可能感兴趣的:(codeforcess刷题记录)