【BZOJ 4326】NOIP2015运输计划

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<stack>
#define depth 21
#define N 300005
using namespace std;

int n,m,tot,ans;
int anc[N][22],a[N],b[N],d[N],lca[N],last[N],dep[N],fa[N],sum[N],v[N],dis[N];
stack<int> s;
struct hh
{
    int to,next,w;
}e[N<<1];
int read()
{
    int ret=0;char c=getchar();
    while(!isdigit(c))c=getchar();
    while(isdigit(c)){ret=(ret<<1)+(ret<<3)+c-‘0‘;c=getchar();}
    return ret;
}
void add(int fr,int to,int w)
{
    e[++tot].to=to;
    e[tot].w=w;
    e[tot].next=last[fr];
    last[fr]=tot;
}
void bfs(int root)
{
    int i,j,now;
    s.push(root);
    dep[root]=1;
    for(i=0;i<=depth;i++)
        anc[root][i]=root;
    while(!s.empty())
    {
        now=s.top();s.pop();
        if(now!=root)
            for(i=1;i<=depth;i++)
                anc[now][i]=anc[anc[now][i-1]][i-1];
        for(i=last[now];i;i=e[i].next)
            if(!dep[e[i].to])
            {
                dep[e[i].to]=dep[now]+1;
                anc[e[i].to][0]=now;
                fa[e[i].to]=now; v[e[i].to]=e[i].w;
                dis[e[i].to]=dis[now]+e[i].w;
                s.push(e[i].to);
            }
    }
}
void swim(int &x,int h)
{
    int i;
    for(i=0;h;i++)
    {
        if(h&1) x=anc[x][i];
        h/=2;
    }
}
int get_lca(int x,int y)
{
    int i;
    if(dep[x]>dep[y]) swap(x,y);
    if(dep[x]!=dep[y]) swim(y,dep[y]-dep[x]);
    if(x==y) return x;
    while(true)
    {
        for(i=0;anc[x][i]!=anc[y][i];i++);
        if(i==0) return anc[x][0];
        i--;x=anc[x][i];y=anc[y][i];
    }
    
}
void work(int now)
{
    int i,j;
    for(i=last[now];i;i=e[i].next)
        if(e[i].to!=fa[now])
        {
            work(e[i].to);
            sum[now]+=sum[e[i].to];
        }
}
bool check(int lim)
{
    int i,j,cnt=0,t=0;
    for(i=1;i<=n;i++) sum[i]=0;
    for(i=1;i<=m;i++)
        if(d[i]>lim)
        {
            cnt++;
            t=max(t,d[i]-lim);
            sum[a[i]]++;sum[b[i]]++;
            sum[lca[i]]-=2;
        }
    work(1);
    for(i=1;i<=n;i++)
        if(sum[i]==cnt&&v[i]>=t) return true;
    return false;
    
}

int main()
{
    int i,j,u,v,w,l,r,mid;
    n=read();m=read();
    for(i=1;i<=n-1;i++)
    {
        u=read();v=read();w=read();
        add(u,v,w);add(v,u,w);
    }
    bfs(1);
    for(i=1;i<=m;i++)
    {
        a[i]=read();b[i]=read();
        lca[i]=get_lca(a[i],b[i]);
        d[i]=dis[a[i]]+dis[b[i]]-(dis[lca[i]]<<1);
    }
    l=0,r=d[1];
    for(i=2;i<=m;i++) r=max(r,d[i]);
    ans=99999999;
    while(l<=r)
    {
        mid=(l+r)>>1;
        if(check(mid)) ans=min(ans,mid),r=mid-1;
        else l=mid+1;
    }
    printf("%d",ans);
    return 0;
}