bzoj4326【2015提高】运输计划（二分答案+lca+树上差分前缀和）

这题我哪会啊。。。这题卡常，我们把每个路径的lca求一次就记录下来，每条路径的长度求出来（借助预处理得出的dis数组，dis[i]表示i到根的路径长度,则len[i][j]=dis[i]+dis[j]-2*dis[lca(i,j)]）。然后按每条路径的长度从大到小排序，也许能快些(
让最大值最小问题。。二分答案，每次判定最大值<=mid是否可能。那么对于每条长度大于mid的路径我们都得给他们去边，然而我们只能去一条边，因此必须去他们的公共边。怎么找公共边呢？假设一共有cnt条路径不合法，对于不合法的路径，我们给这条路径上的边+1，表示这条边出现过一次。最后出现了cnt次的边就是公共边。这个操作可以用树剖实现，但可能卡不过去？因为是相当于只有最后一次的全部查询，所以我们可以采用树上差分+前缀和的思想来搞。怎么搞呢？对路径(u,v)上的边+1，则c[u]+1,c[v]+1,c[lca(u,v)]-2.每个点最后的值就是他的子树的所有值加起来。因此我们最后一遍dfs从底向上即可推出所有的值。总复杂度O(n)。我们看有没有可能删掉一条公共边，使得最长路径减去这条边能够<=mid,如果存在，即满足。

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 300010
inline 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;
}
int n,m,h[N],num=0,fa[N][20],dep[N],dis[N],w[N],Log[N];
int mid,cnt,c[N];
bool flag;
struct edge{
    int to,next,val;
}data[N<<1];
struct Data{
    int u,v,t,len;
}a[N];
inline bool cmp(Data x,Data y){
    return x.len>y.len;
}
void dfs(int x){
    for(int i=1;i<=Log[n];++i){
        if(!fa[x][i-1]) continue;
        fa[x][i]=fa[fa[x][i-1]][i-1];
    }
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa[x][0]) continue;
        fa[y][0]=x;dep[y]=dep[x]+1;w[y]=data[i].val;
        dis[y]=dis[x]+w[y];dfs(y);
    }
}
inline int lca(int x,int y){
    if(dep[x]<dep[y]) swap(x,y);
    int t=dep[x]-dep[y];
    for(int i=Log[t];i>=0;--i)
        if(t&(1<<i)) x=fa[x][i];
    if(x==y) return x;
    for(int i=Log[n];i>=0;--i)
        if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
int calc(int x){
    int res=c[x];
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa[x][0]) continue;
        res+=calc(y);if(flag) return res;
    }if(res!=cnt) return res;
    if(a[1].len-w[x]<=mid) flag=1;return res;
}
bool jud(){
    cnt=0;memset(c,0,sizeof(c));flag=0;
    for(int i=1;i<=m;++i){
        if(a[i].len<=mid) break;cnt++;
        c[a[i].u]++;c[a[i].v]++;c[a[i].t]-=2;
    }calc(1);return flag;
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();Log[0]=-1;
    for(int i=1;i<=n;++i) Log[i]=Log[i>>1]+1;
    for(int i=1;i<n;++i){
        int x=read(),y=read(),v=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;data[num].val=v;
        data[++num].to=x;data[num].next=h[y];h[y]=num;data[num].val=v;
    }dfs(1);
    for(int i=1;i<=m;++i){
        a[i].u=read();a[i].v=read();a[i].t=lca(a[i].u,a[i].v);
        a[i].len=dis[a[i].u]+dis[a[i].v]-2*dis[a[i].t];
    }sort(a+1,a+m+1,cmp);
    int l=0,r=a[1].len;
    while(l<=r){
        mid=l+r>>1;
        if(jud()) r=mid-1;else l=mid+1;
    }printf("%d\n",r+1);
    return 0;
}