求hack(内带解释)
查看原帖
求hack(内带解释)
947881
liuyize549330楼主2025/1/18 20:37

将可以直接到达目标城市的答案(cnt)设置成1,由该点往外扩,每次更新队列。

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int N=1e5+10;
const int M=2e5+10;
int n,m,k;
int a;
int cnt[N];
bool pd[N];
struct re{
	int to;
	int next;
}e[M*2];
int h[N];
int idx;
struct ee{
	int num;
	int from;
}; 
queue<ee> q;
void add(int x,int y){
	e[++idx].to=y;
	e[idx].next=h[x];
	h[x]=idx;
}
signed main(){
	cin>>n>>m>>k;
	memset(h,-1,sizeof h);
	memset(cnt,-1,sizeof cnt);
	for(int i=1;i<=k;i++) cin>>a,pd[a]=1;
	for(int i=1;i<=m;i++){
		int x,y;
		cin>>x>>y;
		if(x==y) continue;
		if(pd[x]) cnt[y]=1,q.push((ee){y,x}); 
		if(pd[y]) cnt[x]=1,q.push((ee){x,y});
		add(x,y);
		add(y,x);
	}
	while(!q.empty()){
		int x=q.front().num;
		int from=q.front().from;
		q.pop();
		for(int i=h[x];i!=-1;i=e[i].next){
			int to=e[i].to;
			if(to!=from&&(cnt[to]==-1||cnt[to]+1<cnt[x])){
				cnt[to]=cnt[x]+1;
				q.push((ee){to,x});
			}
		}
	}
	for(int i=1;i<=n;i++) cout<<cnt[i]<<" ";
	return 0;
}
2025/1/18 20:37
加载中...