E. Equidistant 换根dp

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

来源：CSDN

作者：ccsu_Tomcat

链接：https://blog.csdn.net/qq_44335608/article/details/103027629