[BZOJ4326][NOIP2015][倍增][二分][差分序列]运输计划

Description
公元 2044 年，人类进入了宇宙纪元。L 国有 n 个星球，还有 n−1 条双向航道，每条航道建立在两个星球之间，这 n−1 条航道连通了 L 国的所有星球。小 P 掌管一家物流公司， 该公司有很多个运输计划，每个运输计划形如：有一艘物流飞船需要从 ui 号星球沿最快的宇航路径飞行到 vi 号星球去。显然，飞船驶过一条航道是需要时间的，对于航道 j，任意飞船驶过它所花费的时间为 tj，并且任意两艘飞船之间不会产生任何干扰。为了鼓励科技创新， L 国国王同意小 P 的物流公司参与 L 国的航道建设，即允许小P 把某一条航道改造成虫洞，飞船驶过虫洞不消耗时间。在虫洞的建设完成前小 P 的物流公司就预接了 m 个运输计划。在虫洞建设完成后，这 m 个运输计划会同时开始，所有飞船一起出发。当这 m 个运输计划都完成时，小 P 的物流公司的阶段性工作就完成了。如果小 P 可以自由选择将哪一条航道改造成虫洞， 试求出小 P 的物流公司完成阶段性工作所需要的最短时间是多少？
Input
第一行包括两个正整数 n,m，表示 L 国中星球的数量及小 P 公司预接的运输计划的数量，星球从 1 到 n 编号。接下来 n−1 行描述航道的建设情况，其中第 i 行包含三个整数 ai,bi 和 ti，表示第 i 条双向航道修建在 ai 与 bi 两个星球之间，任意飞船驶过它所花费的时间为 ti。数据保证 1≤ai,bi≤n 且 0≤ti≤1000。接下来 m 行描述运输计划的情况，其中第 j 行包含两个正整数 uj 和 vj，表示第 j 个运输计划是从 uj 号星球飞往 vj号星球。数据保证 1≤ui,vi≤n
Output
输出文件只包含一个整数，表示小 P 的物流公司完成阶段性工作所需要的最短时间。
Sample Input
6 3
1 2 3
1 6 4
3 1 7
4 3 6
3 5 5
3 6
2 5
4 5
Sample Output
11
HINT
将第 1 条航道改造成虫洞： 则三个计划耗时分别为：11,12,11，故需要花费的时间为 12。
将第 2 条航道改造成虫洞： 则三个计划耗时分别为：7,15,11，故需要花费的时间为 15。
将第 3 条航道改造成虫洞： 则三个计划耗时分别为：4,8,11，故需要花费的时间为 11。
将第 4 条航道改造成虫洞： 则三个计划耗时分别为：11,15,5，故需要花费的时间为 15。
将第 5 条航道改造成虫洞： 则三个计划耗时分别为：11,10,6，故需要花费的时间为 11。
故将第 3 条或第 5 条航道改造成虫洞均可使得完成阶段性工作的耗时最短，需要花费的时间为 11。

考试的时候想到了正解的算法，然而不会实现TAT
可以二分答案，对于时间限制lim,找到所有路径长度大于lim的运输计划，显然的，虫洞应选择放在这些计划的交边中权值最大的边。如果删掉这条边后所有计划满足lim，则答案合理。
而难点则在于如何找出交边。
还记得NOIP2012借教室么，区间修改，只需sum[L]+=val,sum[R+1]-=val。
而这里在树上我们也可以运用类似的差分，对于一条路径，sum[u]++,sum[v]++,sum[lca(u,v)]-=2,前缀和扫完后sum等于超过lim的计划总数的边即为交边。
不过这道题要卡常数，表示我连读入优化都写不动啊。
卡常数的出题人都应该被烧死。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=3e5+5,D=16;
int n,m,dep[maxn],to[maxn<<1],worth[maxn<<1],Next[maxn<<1],head[maxn];
int fa[maxn][D+2],dis[maxn][D+2],sum[maxn],edge,cnt,hole,ans,INF;
struct Route
{
    int s,t,lca,len;
    bool operator<(const Route &x)const
    {
        return len<x.len;
    }
}route[maxn];
#define s(x) route[x].s
#define t(x) route[x].t
#define lca(x) route[x].lca
#define len(x) route[x].len
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
void edge_add(int u,int v,int w)
{
    to[++edge]=v;
    worth[edge]=w;
    Next[edge]=head[u];
    head[u]=edge;
}
void dfs(int u)
{

    for (int E=head[u];E;E=Next[E])
    {
        int v=to[E];
        if (v==fa[u][0]) continue;
        fa[v][0]=u;
        dep[v]=dep[u]+1;
        dis[v][0]=worth[E];
        dfs(v);
    }
}
int LCA(int u,int v)
{
    if (dep[u]<dep[v]) swap(u,v);
    int delta=dep[u]-dep[v];
    for (int i=0;i<=D;i++) if (delta&(1<<i)) u=fa[u][i];
    if (u==v) return v;
    for(int i=D;i>=0;i--) if (fa[u][i]!=fa[v][i]) u=fa[u][i],v=fa[v][i];
    return fa[u][0]; 
}
int path(int node)
{
    int ret=0;
    int u=s(node),v=t(node),delta1=dep[u]-dep[lca(node)],delta2=dep[v]-dep[lca(node)];
    for (int i=0;i<=D;i++)
    {
        if (delta1&(1<<i)) ret+=dis[u][i],u=fa[u][i];
        if (delta2&(1<<i)) ret+=dis[v][i],v=fa[v][i];
    } 
    return ret;
}
void update(int u)
{
    for (int E=head[u];E;E=Next[E])
    {
        int v=to[E];
        if (v==fa[u][0]) continue;
        update(v);
        sum[u]+=sum[v];
        if (sum[v]==cnt) hole=max(hole,dis[v][0]);
    }
}
bool check(int lim)
{
    memset(sum,0,sizeof(sum));
    cnt=hole=0;
    for (int i=m;i>=1;i--)
    if (len(i)>lim) cnt++,sum[s(i)]++,sum[t(i)]++,sum[lca(i)]-=2;
    else break;
    if (cnt) update(1);
    return len(m)-hole<=lim;
}
int main()
{
    freopen("transport.in","r",stdin);
    freopen("transport.out","w",stdout);
    n=read(); m=read();
    for (int i=1;i<n;i++)
    {
        int u,v,w;
        u=read(); v=read(); w=read(); 
        edge_add(u,v,w); edge_add(v,u,w);
    }
    dep[1]=1;
    dfs(1);
    for (int i=1;i<=D;i++)
      for (int u=1;u<=n;u++)
      {
        fa[u][i]=fa[fa[u][i-1]][i-1];
        dis[u][i]=dis[u][i-1]+dis[fa[u][i-1]][i-1];
      }
    for (int i=1;i<=m;i++)
    {
        s(i)=read(); t(i)=read();
        lca(i)=LCA(s(i),t(i));
        len(i)=path(i);
        INF=max(INF,len(i));
    }
    sort(route+1,route+m+1); 
    int L=1,R=INF;
    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;
}