[二分+差分]BZOJ 4326——NOIP2015 运输计划

题目梗概

给出一棵有边权的树，并有m个从x到y的路径。

将一条边的边权改为0，使所有路径的最大值最小。

解题思路

因为要使最大值最小，所以我们二分答案。

对于大于答案的路径，我们要修改它。

那么就是把这些路径的交集中最大的边改为0。

求路径交集差分一下就可以了，在差分的时候只需要LCA就可以了。

#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
inline char nc(){
    static char buf[100000],*l=buf,*r=buf;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    if (l==r) return EOF;return *l++;
}
inline int _read(){
    int num=0;char ch=nc();
    while(ch<'0'||ch>'9') ch=nc();
    while(ch>='0'&&ch<='9') num=num*10+ch-48,ch=nc();
    return num;
}
const int maxn=300005;
int tot,son[maxn*2],w[maxn*2],nxt[maxn*2],lnk[maxn],lg;
int dis[maxn],f[maxn][20],n,m,dep[maxn],X[maxn],Y[maxn],c[maxn],fa[maxn],L,R,mid,tag[maxn],sum,num;
inline void add(int x,int y,int z){nxt[++tot]=lnk[x];lnk[x]=tot;son[tot]=y;w[tot]=z;}
inline int max(int x,int y){if (x>y) return x;return y;}
void DFS(int x,int fa){
    dep[x]=dep[fa]+1;f[x][0]=fa;
    for (int j=lnk[x];j;j=nxt[j]) if (son[j]!=fa){
        dis[son[j]]=dis[x]+w[j];
        DFS(son[j],x);
    }
}
void work(){
    for (int j=1;j<=lg;j++)
    for (int i=1;i<=n;i++)
    f[i][j]=f[f[i][j-1]][j-1];
}
int LCA(int x,int y){
    if (dep[x]<dep[y]) swap(x,y);
    for (int i=lg;i>=0;i--) if (dep[f[x][i]]>=dep[y]) x=f[x][i];
    if (x==y) return x;
    for (int i=lg;i>=0;i--) if (f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
    return f[x][0];
}
void query(int x,int fa,int lst){
    for (int j=lnk[x];j;j=nxt[j]) if (son[j]!=fa) query(son[j],x,w[j]),tag[x]+=tag[son[j]];
    if (tag[x]==sum) num=max(num,lst);
}
bool check(int x){
    memset(tag,0,sizeof(tag));int now=0;sum=0;num=0;
    for (int i=1;i<=m;i++)if (c[i]>x) tag[X[i]]++,tag[Y[i]]++,tag[fa[i]]-=2,now=max(now,c[i]),sum++;
    query(1,0,0);
    return now-num<=x;
}
int main(){
    freopen("exam.in","r",stdin);
    freopen("exam.out","w",stdout);
    n=_read();m=_read();lg=log2(n);
    for (int i=1;i<n;i++){int x=_read(),y=_read(),z=_read();add(x,y,z);add(y,x,z);}
    DFS(1,0);work();
    for (int i=1;i<=m;i++){
        X[i]=_read();Y[i]=_read();
        fa[i]=LCA(X[i],Y[i]);c[i]=dis[X[i]]+dis[Y[i]]-2*dis[fa[i]];
        R=max(R,c[i]);
    }
    while(L<=R){
        mid=L+(R-L>>1);
        if (check(mid)) R=mid-1;else L=mid+1;
    }
    printf("%d\n",L);
    return 0;
}