E. Equidistant
题意:求树上是否存在一个点,使得m个给定的点到该点的距离相等。
题解:我设每个点对应两个状态,一个是最远到达m各给定点的距离,一个是最近到达m个给定点的距离,如果mx[u]==mn[u]说明该点合法。
#include<bits/stdc++.h>
#define mk make_pair
#define pi pair<int,int>
using namespace std;
const int maxn = 2e5+10;
int vis[maxn];
vector<int>G[maxn];
vector<pi>sf[maxn];
int mx[maxn],mn[maxn];
const int inf = 1e9+7;
int ans ;
void dfs(int u,int fa)
{
if(vis[u]) mx[u] = mn[u] = 0;
else mx[u] = -inf , mn[u] = inf;
for(auto v:G[u])
if(v!=fa)
{
dfs(v,u);
mx[u] = max(mx[u],mx[v]+1);
mn[u] = min(mn[u],mn[v]+1);
}
}
void dfs2(int u,int fa)
{
if(ans) return ;
if(vis[u]) mx[u] = mn[u] = 0;
else mx[u] = -inf , mn[u] = inf;
sf[u].push_back(mk(mx[u],mn[u]));
for(auto v:G[u])
{
mx[u] = max(mx[u],mx[v]+1);
mn[u] = min(mn[u],mn[v]+1);
sf[u].push_back(mk(mx[u],mn[u]));
}
if(mx[u]==mn[u]){
ans = u;return ;
}
int Mx = -inf , Mn = inf;
for(int i=G[u].size()-1;i>=0;i--)
{
int v = G[u][i];
mx[u] = max(sf[u][i].first,Mx);
mn[u] = min(sf[u][i].second,Mn);
Mx = max(Mx,mx[v]+1);
Mn = min(Mn,mn[v]+1);
if(v!=fa) dfs2(v,u);
}
}
int main()
{
int n,m;scanf("%d%d",&n,&m);
int u,v;
for(int i=1;i<n;i++)
{
scanf("%d%d",&u,&v);
G[u].push_back(v);
G[v].push_back(u);
}
int p;
for(int i=1;i<=m;i++)
{
scanf("%d",&p);
vis[p]++;
}
dfs(1,0);
dfs2(1,0);
if(ans) printf("YES\n%d\n",ans);
else puts("NO");
}