noip2015 运输计划

bzoj传送门
COGS传送门

二分答案比较容易想到，但是对于判断不太好进行，所以要使用新知识：树上差分。
对于二分出的时间，我们需要判断所有的运输计划是否在添加一个虫洞以后都达到要求，所以我们首先将所有计划按照距离排序，将大于要求的计划在树上进行差分，然后统计是否有边在所有计划中，并取其中最长的一条，判断如果将其改成虫洞是否达到要求，并更新答案。
说得比较啰嗦，看不懂就看这一篇文章吧。

吐槽：cogs最后一个点极其卡常，使用各种inline和register才卡过。

CODE：

#include<cstdio>
#include<algorithm>
using namespace std;
const int N=3e5+10;
const int INF=1e9;
struct edge
{
    int nxt,to,dis;
}a[N<<1];
struct work
{
    int s,t,dis,lca;
    inline bool operator <(const work other)const
    {
        return dis>other.dis;
    }
}w[N];
int head[N],f[N],deep[N],size[N],son[N],top[N],dis[N],sum[N];
int n,m,x,y,z,num,tot,L,R,mid,ans,maxcost;
inline int max(const int &a,const int &b){return a>b?a:b;}
inline int min(const int &a,const int &b){return a<b?a:b;}
inline void swap(int &a,int &b){a^=b,b^=a,a^=b;}
inline void read(int &n)
{
    n=0;char c=getchar();
    while(c<'0'||c>'9') c=getchar();
    while(c>='0'&&c<='9') n=n*10+c-48,c=getchar();
}
inline void add(int x,int y,int z)
{
    a[++num].nxt=head[x],a[num].to=y,a[num].dis=z,head[x]=num;
    a[++num].nxt=head[y],a[num].to=x,a[num].dis=z,head[y]=num;
}
inline void dfs(int now)
{
    size[now]=1;
    int tmp=-INF;
    for(register int i=head[now];i;i=a[i].nxt)
      if(a[i].to!=f[now])
      {
        int v=a[i].to;
        f[v]=now;
        deep[v]=deep[now]+1;
        dis[v]=dis[now]+a[i].dis;
        dfs(v);
        size[now]+=size[v];
        if(size[v]>tmp) tmp=size[v],son[now]=v;
      }
}
inline void dfs2(int now,int high)
{
    top[now]=high;
    if(son[now]) dfs2(son[now],high);
    for(register int i=head[now];i;i=a[i].nxt)
      if(a[i].to!=f[now]&&a[i].to!=son[now]) dfs2(a[i].to,a[i].to);
}
inline int LCA(int x,int y)
{
    while(top[x]!=top[y])
      if(deep[top[x]]>deep[top[y]]) x=f[top[x]];
      else y=f[top[y]];
    return deep[x]<deep[y]?x:y;
}
inline int calc(int now)
{
    for(register int i=head[now];i;i=a[i].nxt)
      if(a[i].to!=f[now]) sum[now]+=calc(a[i].to);
    if(sum[now]==tot) maxcost=max(maxcost,dis[now]-dis[f[now]]);
    int ans=sum[now];sum[now]=0;
    return ans;
}
inline bool check(int dist)
{
    tot=m;
    for(register int i=1;i<=m;i++)
      if(w[i].dis>dist) tot=i,sum[w[i].s]++,sum[w[i].t]++,sum[w[i].lca]-=2;
    maxcost=0;
    calc(1);
    if(w[1].dis-maxcost<=dist) return 1;
    return 0;
}
int main()
{
    read(n),read(m);
    for(register int i=1;i<n;i++)
      read(x),read(y),read(z),add(x,y,z);
    dfs(1),dfs2(1,1);
    for(register int i=1;i<=m;i++)
    {
        read(w[i].s),read(w[i].t);
        w[i].lca=LCA(w[i].s,w[i].t);
        w[i].dis=dis[w[i].s]+dis[w[i].t]-(dis[w[i].lca]<<1);
        R=max(R,w[i].dis);
    }
    sort(w+1,w+m+1);
    while(L<=R)
    {
        int mid=(L+R)>>1;
        if(check(mid)) ans=mid,R=mid-1;
        else L=mid+1;
    }
    printf("%d",ans);
    return 0;
}