Codeforces Round #328 (Div. 2) 592 D. Super M 树的直径

题意:有个国家有n个城市，有n-1条道路（双向。两两城市都可以互相到达（其实就是一棵树，然后现在有m城市被攻击.现在有个超级英雄要赶到所有被攻击的城市，他现在在其他国家，有个瞬间转移可以刚开始转移到任何一个城市。问你最小的步数赶到所有被攻击的城市
思路:呃… 好吧..就是因为不补题..所以没见过这种很基础的题
这里我们可以利用深搜去除没有用的城市 求出最小树
这棵树是这样的 每个叶子节点都是要赶去的城市
然后我们要赶去所有的城市步数这么求
因为每个叶子结点都是要赶去的 赶过去过后再返回根节点的步数就是
最小树的边的个数*2
去掉返回根节点的步数就是
最小树的边的个数*2-树的直径
树的直径就是这棵树最长的一个树枝
可以通过两次广搜求
第一次广搜可以搜出来直径的一个端点 第二次搜出来另一个端点 OK AC

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<queue>
#include<stack>
#include<string>
#include<vector>
#include<map>
#include<set>
using namespace std;
#define rfor(i,a,b) for(i=a;i<=b;++i)
#define lfor(i,a,b) for(i=a;i>=b;--i)
#define sfor(i,a,h) for(i=h[a];i!=-1;i=e[i].next)
#define mem(a,b) memset(a,b,sizeof(a))
#define mec(a,b) memcpy(a,b,sizeof(b))
#define cheak(i) printf("%d ",i)
#define min(a,b) (a>b?b:a)
#define max(a,b) (a>b?a:b)
#define inf 0x3f3f3f3f
#define lowbit(x) (x&(-x))
typedef long long LL;
#define maxn 1000005
#define maxm maxn*maxn
#define lson(x) (splay[x].son[0])
#define rson(x) (splay[x].son[1])
vector<int>vec[maxn],vec1[maxn];
int mark[maxn];
int det(int u,int fa,int k)
{
    if(mark[u]) k=1;
    int a=vec[u].size(),i;
    rfor(i,0,a-1)
    {
        int v=vec[u][i];
        if(v!=fa)
        {
            int x=det(v,u,0);
            k|=x;
            if(x)
            {
                vec1[u].push_back(v);
                vec1[v].push_back(u);
            }
        }
    }
    return k;
}
int len;
int dis[maxn],vis[maxn];
void bfs(int sta,int &ed)
{
    int i;
    mem(dis,0);mem(vis,0);
    queue<int>q;
    q.push(sta);
    ed=sta;
    dis[sta]=0;vis[sta]=1;
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        int a=vec[u].size();
        rfor(i,0,a-1)
        if(!vis[vec[u][i]])
        {
            int v=vec[u][i];
            vis[v]=1;
            dis[v]=dis[u]+1;
            q.push(v);
            if(len<dis[v])
            {
                len=dis[v];
                ed=v;
            }
            if(dis[v]==len)
            ed=min(ed,v);
        }
    }
}
int main()
{
    int n,m,i,j,x,u,v;
    scanf("%d%d",&n,&m);
    rfor(i,1,n-1)
    {
        scanf("%d%d",&u,&v);
        vec[u].push_back(v);
        vec[v].push_back(u);
    }
    rfor(i,1,m)
    {
        scanf("%d",&x);
        mark[x]=1;
    }
    det(x,0,0);
    rfor(i,1,n) vec[i].clear();
    int ans=0;
    rfor(i,1,n)
    {
        int a=vec1[i].size();
        rfor(j,0,a-1)
        {
            vec[i].push_back(vec1[i][j]);
            ans++;
        }
    }
    int sta,ed;
    bfs(x,sta),bfs(sta,ed);
    printf("%d\n%d\n",min(sta,ed),ans-len);
    return 0;
}