CodeForces 592D Super M

先把没用的边去掉，求出包含m个点的最小树。然后求出最小树的直径就可以得到答案了。

#include <cstdio>
#include <cstring>
#include <vector>
#include<cmath>
using namespace std;

const int maxn=;
struct Edge
{
    int u,v;
    int next;
} e[*maxn];
int tot;
int h[maxn];
bool f[maxn],q[maxn];
int head[maxn];
int n,m;

int st1,st2;
int len;

void add(int a,int b)
{
    e[tot].u=a;
    e[tot].v=b;
    e[tot].next=head[a];
    head[a]=tot;
    tot++;
}

void dfs(int x)
{
    f[x]=;
    for(int i=head[x]; i!=-; i=e[i].next)
    {
        if(f[e[i].v]==)
        {
            dfs(e[i].v);
            if(q[e[i].v]==) q[x]=;
        }
    }
}

void Find(int x,int le,int op)
{
    f[x]=;
    if(op==)
    {
        if(le>len) len=le,st1=x;
        else if(le==len&&x<st1) st1=x;
    }
    else if(op==)
    {
        if(le>len) len=le,st2=x;
        else if(le==len&&x<st2) st2=x;
    }

    for(int i=head[x]; i!=-; i=e[i].next)
    {
        if(q[e[i].u]==) continue;
        if(q[e[i].v]==) continue;
        if(f[e[i].v]==) Find(e[i].v,le+,op);
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    memset(f,,sizeof f);
    memset(q,,sizeof q);
    memset(head,-,sizeof head);
    tot=;
    for(int i=; i<n; i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        add(u,v);
        add(v,u);
    }
    for(int i=; i<=m; i++)
    {
        int x;
        scanf("%d",&x);
        q[x]=;
    }
    int g;
    for(int i=; i<=n; i++) if(q[i]==)
        {
            g=i;
            dfs(g);
            break;
        }

    st1=st2=0x7FFFFFFF;
    memset(f,,sizeof f);
    len=;
    Find(g,,);
    memset(f,,sizeof f);
    len=;
    Find(st1,,);

    int ans1=min(st1,st2);

    int ans2=;
    for(int i=; i<tot; i=i+)
    {
        if(q[e[i].u]==) continue;
        if(q[e[i].v]==) continue;
        ans2=ans2+;
    }
    ans2=ans2-len;
    printf("%d\n%d\n",ans1,ans2);
    return ;
}