codevs 1218 疫情控制

啊好烦这道题。。。。

基本思路网上都有。

注意的一点是在匹配的时候，如果有军队的来源没有被匹配到，那么就先匹配这个来源。（因为不花钱）。

不过数据好水。。。。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxv 50050
#define maxe 100500
using namespace std;
int n,x,y,z,g[maxv],nume=,m,c[maxv],anc[maxv][],dis[maxv][],l=,r=,disr[maxv],kr=,fath[maxv];
int top=,cnt=;
bool vis[maxv],ret[maxv];
struct edge
{
    int v,w,nxt;
}e[maxe];
struct status
{
    int val,id;
}regis[maxv],stack[maxv];
bool cmp(status a,status b)
{
    return a.val<b.val;
}
void addedge(int u,int v,int w)
{
    e[++nume].v=v;
    e[nume].w=w;
    e[nume].nxt=g[u];
    g[u]=nume;
}
void dfs1(int x,int father)
{
    if (father==) fath[x]=x;
    else if (x!=) fath[x]=fath[father];
    for (int i=g[x];i;i=e[i].nxt)
    {
        int v=e[i].v;
        if (v!=father)
        {
            anc[v][]=x;dis[v][]=e[i].w;
            disr[v]=disr[x]+e[i].w;r=max(r,disr[v]);
            dfs1(v,x);
        }
    }
}
bool dfs2(int x,int father)
{
    if (vis[x]) return true;
    int cnt1=,cnt2=;
    for (int i=g[x];i;i=e[i].nxt)
    {
        int v=e[i].v;
        if (v!=father)
        {
            cnt1++;
            if (dfs2(v,x)) cnt2++;
        }
    }
    if ((cnt1==cnt2) && (cnt1!=)) return true;
    return false;
}
void get_table()
{
    r+=kr;
    for (int i=;i<=;i++)
        for (int j=;j<=n;j++)
        {
            anc[j][i]=anc[anc[j][i-]][i-];
            dis[j][i]=dis[j][i-]+dis[anc[j][i-]][i-];
        }
}
int get_pnt(int x,int r)
{
    for (int i=;i>=;i--)
    {
        if (r>=dis[x][i])
        {
            r-=dis[x][i];
            x=anc[x][i];
        }
    }
    return x;
}
bool check(int x)
{
    memset(vis,false,sizeof(vis));
    memset(ret,false,sizeof(ret));
    top=;cnt=;
    for (int i=;i<=m;i++)
    {
        int v=c[i];
        if (disr[v]<x)
        {
            top++;
            stack[top].val=x-disr[v];
            stack[top].id=fath[v];
        }
        else if (disr[v]==x) vis[fath[v]]=true;
        else vis[get_pnt(v,x)]=true;
    }
    for (int i=g[];i;i=e[i].nxt)
    {
        int v=e[i].v;
        if (!dfs2(v,))
        {
            regis[++cnt].val=disr[v];
            regis[cnt].id=v;
            ret[v]=true;
        }
    }
    if (top<cnt) return false;
    sort(stack+,stack+top+,cmp);
    sort(regis+,regis+cnt+,cmp);
    int p1=,p2=;
    while  ((p2<=cnt) && (p1<=top))
    {
        if (ret[stack[p1].id]) {ret[stack[p1].id]=false;p1++;}
        else
        {
            while ((!ret[regis[p2].id]) && (p2<=cnt)) p2++;
            while ((stack[p1].val<regis[p2].val) && (p1<=top)) p1++;
            if (p2==cnt+) break;
            if (p1==top+) break;
            ret[regis[p2].id]=false;p1++;p2++;
        }
    }
    for (int i=g[];i;i=e[i].nxt)
    {
        int v=e[i].v;
        if (ret[v]) return false;
    }
    return true;
}
int get_ans()
{
    int ans;
    while (l<=r)
    {
        int mid=(l+r)>>;
        if (check(mid)) {ans=mid;r=mid-;}
        else l=mid+;
    }
    return ans;
}
int main()
{
    scanf("%d",&n);
    for (int i=;i<=n-;i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        addedge(x,y,z);
        addedge(y,x,z);
        if ((x==) || (y==)) kr=max(kr,z);
     }
    scanf("%d",&m);
    for (int i=;i<=m;i++)
        scanf("%d",&c[i]);
    dfs1(,);
    get_table();
    printf("%d\n",get_ans());
    return ;
}