【BZOJ】4326 NOIP2015 运输计划 二分+LCA+树上差分

题目传送门

有了之前的铺垫，突然发现这题还是挺水的诶。（逃）

题目要求最大时间的最小值，想到了二分，但是怎么去判断二分出来的值呢？

我们开两个数组$sum$和$v$，$sum$数组表示每条边的标记值，$v$表示每条边的权值。我们把这两个本来是边的值转化到每条边深度较大的节点上，这个显然没有对答案产生影响，并且可以简化代码。

现在我们要判断二分出来的值$mid$，显然路径的权值和$w \leq mid$的路径不需要考虑。对于权值和$w > mid$的路径，我们在这条路径的起点$st$和$ed$的$sum$数组上 打上$+1$的标记，在$lca(st,ed)$的$sum$数组上打上$-1$的标记。

在所有边都打过标记后，遍历一次整棵树，算出所有非根节点的标记$sum$，也就是每条边的标记。如果一条边的标记等于权值和$w > mid$的路径的条数，并且这条边的权值大于$max(w-mid)$，说明修改这条边可以使所有路径的权值和都小于等于$mid$，也就是这个$mid$满足成为答案的条件，那么二分的右边界$r=mid-1$；否则二分的左边界$l=mid+1$。

然后这题就被愉快的秒掉啦。

附上AC代码：

#include <cstdio>
#include <cctype>
#include <algorithm>
#include <cstring>
using namespace std;

const int N=3e5+10;
struct side{
    int to,w,nt;
}s[N<<1];
int n,m,x,y,w,num,h[N],st[N],ed[N],lca[N],l,r,mid,cost[N],sum[N];
int sz[N],f[N],v[N],d[N],hs[N],top[N],dis[N];

inline char nc(){
    static char ch[100010],*p1=ch,*p2=ch;
    return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;
}

inline void read(int &a){
    static char c=nc();int f=1;
    for (;!isdigit(c);c=nc()) if (c=='-') f=-1;
    for (a=0;isdigit(c);a=a*10+c-'0',c=nc());
    a*=f;return;
}

inline void add(int x,int y,int w){
    s[++num]=(side){y,w,h[x]},h[x]=num;
    s[++num]=(side){x,w,h[y]},h[y]=num;
}

inline void so1(int x,int fa){
    d[x]=d[f[x]=fa]+1,sz[x]=1;
    for (int i=h[x]; i; i=s[i].nt)
        if (s[i].to!=fa){
            dis[s[i].to]=dis[x]+(v[s[i].to]=s[i].w);
            so1(s[i].to,x),sz[x]+=sz[s[i].to];
            if (sz[s[i].to]>sz[hs[x]]) hs[x]=s[i].to;
        }
    return;
}

inline void so2(int x,int fa){
    top[x]=fa;
    if (hs[x]) so2(hs[x],fa);
    for (int i=h[x]; i; i=s[i].nt)
        if (s[i].to!=f[x]&&s[i].to!=hs[x]) so2(s[i].to,s[i].to);
    return;
}

inline int query(int x,int y){
    for (int fx=top[x],fy=top[y]; fx!=fy; x=f[fx],fx=top[x])
        if (d[fx]<d[fy]) swap(fx,fy),swap(x,y);
    return d[x]<d[y]?x:y;
}

inline void so(int x){
    for (int i=h[x]; i; i=s[i].nt)
        if (s[i].to!=f[x]) so(s[i].to),sum[x]+=sum[s[i].to];
    return;
}

inline bool check(int mid){
    int c=0,mx=0;memset(sum,0,sizeof sum);
    for (int i=1; i<=m; ++i)
        if (cost[i]>mid){
            ++sum[st[i]],++sum[ed[i]],sum[lca[i]]-=2;
            mx=max(mx,cost[i]-mid),++c;
        }
    so(1);
    for (int i=2; i<=n; ++i) if (sum[i]==c&&v[i]>=mx) return 1;
    return 0;
}

int main(void){
    read(n),read(m);
    for (int i=1; i<n; ++i) read(x),read(y),read(w),add(x,y,w);
    so1(1,0),so2(1,1);
    for (int i=1; i<=m; ++i){
        read(st[i]),read(ed[i]),lca[i]=query(st[i],ed[i]);
        r=max(r,cost[i]=dis[st[i]]+dis[ed[i]]-(dis[lca[i]]<<1));
    }
    while (l<=r) if (check(mid=l+r>>1)) r=mid-1; else l=mid+1;
    return printf("%d\n",l),0;
}