【二分+LCA+差分】BZOJ4326(NOIP2015)[运输计划]题解

题目概述

给 $n$ 个点的树和 $m$ 个运输计划( $x\to y$ )，完成所有运输计划的时间是运输计划中的最大时间。现在可以把一条边的边权改为 $0$ ，求完成所有运输计划的最快时间。

解题报告

由于求最大值的最小值，所以想到二分枚举答案 $mid$ 。

然后预先求出LCA算出每个运输计划的时间，我们就知道有多少个运输计划超出了 $mid$ 。

对于每个超出 $mid$ 的运输计划，我们肯定需要把这个运输计划路径上的一条边改为 $0$ 。如果一条边在所有运输计划的路径上，那么这条边就是可选的，从所有可选的边中选出最大的边再验证是否可行即可。

怎么知道一条边在所有运输计划的路径上？我们可以树链剖分路径加，但是太慢（这道题好像 $n=300000$ ）。所以我们用差分就可以方便解决了。

示例程序

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=300000,maxm=300000,Log=19;

int n,m,dis[maxn+5],fa[maxn+5][Log+5],dep[maxn+5],v[maxn+5];
int E,lnk[maxn+5],nxt[maxn*2+5],son[maxn*2+5],w[maxn*2+5];
int s[maxm+5],t[maxm+5],lca[maxm+5],cst[maxm+5],sum[maxn+5];

#define Eoln(x) ((x)==10||(x)==13||(x)==EOF)
inline char readc()
{
    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 readi(int &x)
{
    int tot=0,f=1;char ch=readc(),lst='+';
    while ('9'<ch||ch<'0') {if (ch==EOF) return EOF;lst=ch;ch=readc();}
    if (lst=='-') f=-f;
    while ('0'<=ch&&ch<='9') tot=(tot<<3)+(tot<<1)+ch-48,ch=readc();
    return x=tot*f,Eoln(ch);
}
inline void Add(int x,int y,int z) {son[++E]=y;w[E]=z;nxt[E]=lnk[x];lnk[x]=E;}
void Dfs(int x,int Fa=0)
{
    for (int j=lnk[x];j;j=nxt[j]) if (son[j]!=Fa)
        dis[son[j]]=dis[x]+(v[son[j]]=w[j]),dep[son[j]]=dep[x]+1,fa[son[j]][0]=x,Dfs(son[j],x);
}
void Make()
{
    for (int j=1;j<=Log;j++)
    for (int i=1;i<=n;i++)
        fa[i][j]=fa[fa[i][j-1]][j-1];
}
inline int LCA(int x,int y)
{
    if (dep[x]<dep[y]) swap(x,y);
    for (int j=Log;dep[x]>dep[y]&&j>=0;j--) if (dep[x]-(1<<j)>=dep[y]) x=fa[x][j];
    if (x==y) return x;
    for (int j=Log;j>=0;j--) if (fa[x][j]!=fa[y][j]) x=fa[x][j],y=fa[y][j];
    return fa[x][0];
}
void Count(int x,int fa=0) {for (int j=lnk[x];j;j=nxt[j]) if (son[j]!=fa) Count(son[j],x),sum[x]+=sum[son[j]];}
inline bool check(int mid)
{
    int tot=0,MAX=0;memset(sum,0,sizeof(sum));
    for (int i=1;i<=m;i++)
        if (cst[i]>mid)
        {
            sum[s[i]]++;sum[t[i]]++;sum[lca[i]]-=2;
            MAX=max(MAX,cst[i]-mid);tot++;
        }
    Count(1);for (int i=2;i<=n;i++) if (sum[i]==tot&&v[i]>=MAX) return true;
    return false;
}
int main()
{
    freopen("program.in","r",stdin);
    freopen("program.out","w",stdout);
    readi(n);readi(m);
    for (int i=1,x,y,z;i<n;i++) readi(x),readi(y),readi(z),Add(x,y,z),Add(y,x,z);
    Dfs(1);Make();int L=0,R=0;
    for (int i=1;i<=m;i++)
    {
        readi(s[i]),readi(t[i]);lca[i]=LCA(s[i],t[i]);
        R=max(R,cst[i]=dis[s[i]]-dis[lca[i]]+dis[t[i]]-dis[lca[i]]);
    }
    for (int mid=L+(R-L>>1);L<=R;mid=L+(R-L>>1)) if (check(mid)) R=mid-1; else L=mid+1;
    return printf("%d\n",L),0;
}