noip2012疫情控制

（blockade.cpp/c/pas）
【问题描述】
H国有n个城市，这 n个城市用n-1条双向道路相互连通构成一棵树， 1号城市是首都，
也是树中的根节点。
H 国的首都爆发了一种危害性极高的传染病。当局为了控制疫情，不让疫情扩散到边境
城市（叶子节点所表示的城市） ，决定动用军队在一些城市建立检查点，使得从首都到边境
城市的每一条路径上都至少有一个检查点， 边境城市也可以建立检查点。 但特别要注意的是，
首都是不能建立检查点的。
现在，在 H 国的一些城市中已经驻扎有军队，且一个城市可以驻扎多个军队。一支军队可以在有道路连接的城市间移动，并在除首都以外的任意一个城市建立检查点，且只能在
一个城市建立检查点。 一支军队经过一条道路从一个城市移动到另一个城市所需要的时间等
于道路的长度（单位：小时）。
请问最少需要多少个小时才能控制疫情。注意：不同的军队可以同时移动。
【输入】
输入文件名为blockade.in。
第一行一个整数 n，表示城市个数。
接下来的n-1行，每行 3个整数，u、v、w，每两个整数之间用一个空格隔开，表示从
城市u到城市v有一条长为 w的道路。数据保证输入的是一棵树，且根节点编号为 1。
接下来一行一个整数 m，表示军队个数。
接下来一行m个整数，每两个整数之间用一个空格隔开，分别表示这 m个军队所驻扎
的城市的编号。

【输出】
输出文件为blockade.out。
共一行，包含一个整数，表示控制疫情所需要的最少时间。如果无法控制疫情则输出-1。
【输入输出样例】
blockade.in
4
1 2 1
1 3 2
3 4 3
2
2 2

blockade.out
3
【输入输出样例说明】
第一支军队在2号点设立检查点，第二支军队从2号点移动到3 号点设立检查点，所需
时间为3个小时。

【数据范围】
保证军队不会驻扎在首都。
对于20%的数据，2≤ n≤ 10；
对于40%的数据，2 ≤n≤50，0

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=50005;
int n,num,m;
bool vis[maxn],ms[maxn];
int head[maxn],army[maxn],root[maxn],sons[maxn],con[maxn],r[maxn],w[maxn];
int fa[20][maxn],dis[20][maxn];
struct edge{
    int to,v,next;
}g[maxn];
void adde(int u,int v,int w){
    g[++num].to=v;
    g[num].v=w;
    g[num].next=head[u];
    head[u]=num;
}
void dfs(int x){
    vis[x]=true;
    for(int i=head[x];i!=-1;i=g[i].next){
        if(vis[g[i].to]) continue;
        fa[0][g[i].to]=x;
        dis[0][g[i].to]=g[i].v;
        sons[x]++;
        dfs(g[i].to);
    }
}
void beizen(){
    for(int i=1;i<20;i++){
        for(int j=1;j<=n;j++){
            fa[i][j]=fa[i-1][fa[i-1][j]];
            dis[i][j]=dis[i-1][j]+dis[i-1][fa[i-1][j]];
        }
    }
}
bool check(int de){
    memset(ms,0,sizeof(ms));
    memset(r,0,sizeof(r));
    memset(w,0,sizeof(w));
    int tot=0;
    //printf("%d",1);
    for(int i=1;i<=m;i++){
        if(de>root[i]) tot++,r[tot]=de-root[i];
        else if(de==root[i]){
            int f=army[i];
            for(int j=19;j>=0;j--) 
              if(fa[j][f]!=1) f=fa[j][f];
            ms[f]=true;
        }
        else{
            //printf("%d",root[i]);
            int x=de,f=army[i];
            while (x>0){
                for(int j=19;j>=0;j--) 
                  if(dis[j][f]<=x) {
                  x-=dis[j][f];
                  f=fa[j][f];
                  //printf("%d ",j);
                  }
                if(x-dis[0][f]<0) break;
            }
            //printf("%d ",x);
            //printf("%d %d\n",f,army[i]);
            while (sons[fa[0][f]]==1&&fa[0][f]!=1) f=fa[0][f];
            if(fa[0][f]==1) ms[f]=true; 
            //printf("%d",f);
        }
    }
    //printf("-1");
    int tu=0;
    for(int i=head[1];i!=-1;i=g[i].next){
        if(ms[g[i].to]==false) tu++,w[tu]=g[i].v;
    }
    if(tu>tot) return false;
    sort(r+1,r+tot+1);
    sort(w+1,w+tu+1);
    int j=1;
    for(int i=1;i<=tu;i++){
        while(r[j]<w[i]&&j<=tot) j++;
        if(j>tot) return false;
    }
    return true;
}
int main(){
    memset(head,-1,sizeof(head));
    memset(vis,0,sizeof(vis));
    freopen("blockade.in","r",stdin);
    freopen("blockade.out","w",stdout);
    scanf("%d",&n);
    num=0;
    fa[0][1]=1;
    for(int i=1;i<=n-1;i++){
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        adde(u,v,w);
        adde(v,u,w);
    }
    int tot=0;
    for(int i=head[1];i!=-1;i=g[i].next) tot++;
    scanf("%d",&m);
    if(tot>m) {
        printf("-1");
        return 0;
    }
    dfs(1);
    beizen();
    int maxr=-1;
    for(int i=1;i<=m;i++) scanf("%d",&army[i]);
    for(int i=1;i<=m;i++){
        root[i]=dis[19][army[i]];
        maxr=max(maxr,root[i]);
    }
    //printf("%d",check(9));
    int left=1,right=6*maxr;
    while (left<right){
        int mid=(left+right)/2;
        //printf("m=%d \n",mid);
        if(check(mid)) right=mid-1;
        else left=mid+1;
    }
    printf("%d",right);
    return 0;
}