洛谷P2680 运输计划——树上差分

题目：https://www.luogu.org/problemnew/show/P2680

久违地1A了好高兴啊！

首先，要最大值最小，很容易想到二分；

判断当前的 mid 是否可行，需要看看有没有去掉一条边使满足的方案；

这就需要树上差分来找出每条边被几个超过 mid 的路线覆盖；

若有一条边正好被所有超过 mid 的路线覆盖，且去掉它之后最大的路线也能满足，就是可行的。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int const maxn=3e5+;
int n,m,head[maxn],ct,cnt,st[maxn],ed[maxn],tag[maxn],lca[maxn],len[maxn];
int h[maxn],ans,mx,f[maxn][],dep[maxn],l,r,mid;
bool flag;
struct N{
    int to,next,w;
    N(int t=,int n=,int w=):to(t),next(n),w(w) {}
}edge[maxn<<];
void add(int x,int y,int z){edge[++ct]=N(y,head[x],z); head[x]=ct;}
void init(int x,int fa)
{
    f[x][]=fa; dep[x]=dep[fa]+;
    for(int i=;i<=;i++)f[x][i]=f[f[x][i-]][i-];
    for(int i=head[x],u;i;i=edge[i].next)
        if((u=edge[i].to)!=fa) h[u]=h[x]+edge[i].w, init(u,x);
}
int LCA(int x,int y)
{
    if(dep[x]<dep[y])swap(x,y);
    int k=dep[x]-dep[y];
    for(int i=;i<=;i++)
        if(k&(<<i))x=f[x][i];
    for(int i=;i>=;i--)
        if(f[x][i]!=f[y][i])x=f[x][i],y=f[y][i];
    if(x==y)return x;
    return f[x][];
}
void dfs(int x)
{
    for(int i=head[x],u;i;i=edge[i].next)
    {
        if((u=edge[i].to)==f[x][])continue;
        dfs(u); if(flag)return;
        if(tag[u]==cnt && mx-edge[i].w<=mid)flag=;
        tag[x]+=tag[u];
    }
}
bool ck()
{
    cnt=; flag=;
    memset(tag,,sizeof tag);
    for(int i=;i<=m;i++)
        if(len[i]>mid)
        {
            tag[st[i]]++; tag[ed[i]]++;
            tag[lca[i]]-=; cnt++;
        }
    dfs();
    return flag;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=,x,y,z;i<n;i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z); add(y,x,z);
    }
    init(,);
    for(int i=;i<=m;i++)
    {
        scanf("%d%d",&st[i],&ed[i]);
        lca[i]=LCA(st[i],ed[i]);
        len[i]=h[st[i]]+h[ed[i]]-*h[lca[i]];
        mx=max(mx,len[i]);
    }
    l=,r=mx;
    while(l<=r)
    {
        mid=(l+r)>>;
        if(ck())ans=mid,r=mid-;
        else l=mid+;
    }
    printf("%d",ans);
    return ;
}