中国石油大学秋季组队训练赛 Equidistant （树上多源BFS）

题目链接：点此跳转

题目大意：给定一颗n个结点的树和m个树上的节点，问是否存在一个节点到m个点的距离相等，一条边距离为1，存在的话输出YES和此节点，反之输出NO。

解题思路：因为树上两点的距离为1，bfs每层只能走1，同时要到m个点的距离相等，我们可能想到的是：枚举每个点bfs看能不能更新成功，但是因为n的范围是2e5，所以会t，因此我们可以在m个点多源bfs，记录分别能到的点，然后判定即可。 （一开始一直想的是树上两点的距离，直接往lca上莽了，然后憋了一个多小时，看了题解才知道是BFS，当场脑溢血）

Code：

#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
const int maxn=2e5+7;

struct Edge{
	int to,nex;
}edge[maxn<<1];

int head[maxn],tot;

inline void add(int u,int v){
	edge[++tot]=(Edge){v,head[u]},head[u]=tot;
	edge[++tot]=(Edge){u,head[v]},head[v]=tot;
}

int vis[maxn],dis[maxn],cnt[maxn],n,m;
queue<int> q;

int bfs(){
	while(!q.empty()){
		int x=q.front();q.pop();
		if(cnt[x]==m) return x;
		for(int i=head[x];i;i=edge[i].nex){
			int y=edge[i].to;
			if(dis[y]==-1||dis[y]==dis[x]+1){
				dis[y]=dis[x]+1;
				cnt[y]+=cnt[x];
				if(!vis[y]) vis[y]=1,q.push(y);
			}
		}
	}
	return -1;
}

int main(){
	
	memset(dis,-1,sizeof dis);
	scanf("%d%d",&n,&m);
	
	for(int i=1;i<=n-1;i++){
		int x,y;
		scanf("%d%d",&x,&y);
		add(x,y);
	}
	
	for(int i=1;i<=m;i++){
		int x;
		scanf("%d",&x);
		q.push(x);
		vis[x]=1;
		dis[x]=0;
		cnt[x]=1;
	}
	int ans=bfs();
	if(ans!=-1) printf("YES\n%d\n",ans);
	else puts("NO");
	
	
}