NOIP2012 疫情控制

代码还有点问题，先放这里存档

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<iterator>
#include<algorithm>
using namespace std;

const int maxn=50010;
int n,m;
struct Edge
{
    int u;
    long long w;
}M[maxn],Node[maxn];
vector<Edge> T[maxn];
vector<int> P_n,M_n;
int P[maxn];
long long sum=0;
bool Map[maxn];

void init(int p)
{
    vector<struct Edge>::iterator iter;
    for(iter=T[p].begin();iter!=T[p].end();iter++)
        if(M[iter->u].u==0)
        {
            M[iter->u].u=p;
            M[iter->u].w=iter->w;
            init(iter->u);
        }
        else
        {
            T[p].erase(iter);
            iter--;
        }
}

bool dfs(int a)
{
    if(Map[a]) return true;
    bool F=true;
    for(int i=0;i<T[a].size();i++)
        if(!dfs(T[a][i].u))
            F=false;
    if(T[a].size()==0) F=false;
    return F;
}

bool cmp1(int a,int b)
{
    return Node[a].w<Node[b].w;
}
bool cmp2(int a,int b)
{
    return T[1][a].w<T[1][b].w;
}
bool cmp3(Edge a,Edge b)
{
    if(a.u==b.u) return a.w>b.w;
    else return a.u<b.u;
}

bool check(long long a)
{
    memset(Map,0,sizeof(Map));
    P_n.clear();
    M_n.clear();
    for(int i=0;i<m;i++)
    {
        Node[i].u=P[i];
        Node[i].w=a;
        while(M[Node[i].u].u!=1&&Node[i].w>=M[Node[i].u].w)
        {
            Node[i].w-=M[Node[i].u].w;
            Node[i].u=M[Node[i].u].u;
        }
        Map[Node[i].u]=true;
    }
    sort(Node,Node+m,cmp3);
    for(int i=0;i<m;i++)
    if(M[Node[i].u].u==1)
    {
        if(Node[i].u!=Node[i+1].u) Map[Node[i].u]=false;
        if(Node[i].w>=M[Node[i].u].w&&dfs(Node[i].u))
        {
            Node[i].w-=M[Node[i].u].w;
            Node[i].u=M[Node[i].u].u;
            P_n.push_back(i);
        }
        else Map[Node[i].u]=true;
    }
    for(int i=0;i<T[1].size();i++)
        if(!dfs(T[1][i].u))
            M_n.push_back(i);
    sort(P_n.begin(),P_n.end(),cmp1);
    sort(M_n.begin(),M_n.end(),cmp2);
    int j=0;
    for(int i=0;i<M_n.size();i++)
    {
        if(j>=P_n.size())
            return false;
        while(Node[P_n[j]].w<T[1][M_n[i]].w)
        {
            j++;
            if(j>=P_n.size())
                return false;
        }
        j++;
    }
    return true;
}

int main()
{
    FILE *in,*out;
    in=fopen("1.in","r");
    fscanf(in,"%d",&n);
    for(int i=1;i<n;i++)
    {
        int u,v,w;
        fscanf(in,"%d %d %d",&u,&v,&w);
        T[u].push_back(Edge{v,w});
        T[v].push_back(Edge{u,w});
        M[i].u=0;M[i].w=0;
        sum+=w;
    }
    fclose(in);
    M[n].u=0;M[n].w=0;
    M[1].u=-1;
    scanf("%d",&m);
    for(int i=0;i<m;i++)
        scanf("%d",&P[i]);
    if(T[1].size()>m)
    {
        printf("-1");
        return 0;
    }
    init(1);
    long long L=0,R=sum,Mid;
    while(L<R)
    {
        Mid=(L+R)>>1;
        if(check(Mid)) R=Mid;
        else L=Mid+1;
    }
    out=fopen("1.out","w");
    fprintf(out,"%lld",L);
    fclose(out);
}