[2017集训队作业自选题#117]Monkey and Tree

题目大意

一颗边权树，给你若干点对(ai,bi)。
找到两对点对i和j，使得dis(ai,aj)+dis(bi,bj)最大。

做法

先建立点分树。
然后进行点分，对于当前分治中心x，讨论最终点对中两个a路径是否过x。
不过x，递归处理。
过x，求出当前分治联通块每个点到x的距离，然后接下来按一个一个子树处理（一起询问，再一起插入，来保证不在同一个子树），我们把b插入到点分树中，对于点分树一个分治中心y，处理出dis(a,x)+dis(b,y)的最大值、最大值来自哪个子树以及不来自同一个子树的次大值，即可完成询问。

#include<cstdio>#include<algorithm>#define fo(i,a,b) for(i=a;i<=b;i++)using namespace std;typedef long long ll;const int maxn=100000+10;const ll inf=10000000000000000;int h[maxn],go[maxn*2],dis[maxn*2],nxt[maxn*2],a[maxn];int ask[maxn][2],h2[maxn],g2[maxn],n2[maxn],size[maxn];int belong[maxn][25],son[maxn][25],mxf[maxn];int sta[maxn];ll d[maxn],dep[maxn][25],mx[maxn],se[maxn];bool bz[maxn],pd[maxn];int i,j,k,l,t,n,m,top,tot,cnt;ll ans;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;}void add(int x,int y,int z){    go[++tot]=y;    dis[tot]=z;    nxt[tot]=h[x];    h[x]=tot;}void add2(int x,int y){    g2[++tot]=y;    n2[tot]=h2[x];    h2[x]=tot;}void travel(int x,int y){    a[++top]=x;    size[x]=1;    int t=h[x];    while (t){        if (!bz[go[t]]&&go[t]!=y){            travel(go[t],x);            size[x]+=size[go[t]];        }        t=nxt[t];    }}void dg(int x,int y,int d,int c){    son[x][d]=c;    int t=h[x];    while (t){        if (!bz[go[t]]&&go[t]!=y){            dep[go[t]][d]=dep[x][d]+(ll)dis[t];            dg(go[t],x,d,c);        }        t=nxt[t];    }}void build(int x,int y){    top=0;    travel(x,0);    int i,t,j=x,k=0;    while (1){        t=h[j];        while (t){            if (!bz[go[t]]&&go[t]!=k&&size[go[t]]>top/2){                k=j;                j=go[t];                break;            }            t=nxt[t];        }        if (!t) break;    }    fo(i,1,top) belong[a[i]][y]=j;    t=h[j];    while (t){        if (!bz[go[t]]){            dep[go[t]][y]=dis[t];            dg(go[t],j,y,go[t]);        }        t=nxt[t];    }    son[j][y]=j;    bz[j]=1;    t=h[j];    while (t){        if (!bz[go[t]]) build(go[t],y+1);        t=nxt[t];    }}void change(int x,ll v){    int j;    int y;    ll w;    fo(j,0,25){        y=belong[x][j];        if (!pd[y]){            pd[y]=1;            sta[++cnt]=y;        }        w=v+dep[x][j];        if (mxf[y]==son[x][j]) mx[y]=max(mx[y],w);        else if (w>=mx[y]){            se[y]=mx[y];            mx[y]=w;            mxf[y]=son[x][j];        }        else if (w>=se[y]) se[y]=w;        if (x==y) break;    }}void query(int x,ll v){    int j;    int y;    ll w;    fo(j,0,25){        y=belong[x][j];        if (mx[y]==-inf) return;        w=v+dep[x][j];        if (mxf[y]==son[x][j]) ans=max(ans,w+se[y]);        else ans=max(ans,w+mx[y]);        if (x==y) break;    }}void cx(int x,int y,int d){    int t=h2[x];    while (t){        query(g2[t],dep[x][d]);        t=n2[t];    }    t=h[x];    while (t){        if (!bz[go[t]]&&go[t]!=y) cx(go[t],x,d);        t=nxt[t];    }}void cr(int x,int y,int d){    int t=h2[x];    while (t){        change(g2[t],dep[x][d]);        t=n2[t];    }    t=h[x];    while (t){        if (!bz[go[t]]&&go[t]!=y) cr(go[t],x,d);        t=nxt[t];    }}void solve(int x,int y){    int j=belong[x][y];    cnt=0;    int t=h2[j];    while (t){        query(g2[t],0);        change(g2[t],0);        t=n2[t];    }    t=h[j];    while (t){        if (!bz[go[t]]){            cx(go[t],j,y);            cr(go[t],j,y);        }        t=nxt[t];    }    int i;    fo(i,1,cnt){        pd[sta[i]]=0;        mx[sta[i]]=se[sta[i]]=-inf;        mxf[sta[i]]=0;    }    bz[j]=1;    t=h[j];    while (t){        if (!bz[go[t]]) solve(go[t],y+1);        t=nxt[t];    }}int main(){    n=read();m=read();    fo(i,1,n-1){        j=read();k=read();l=read();        add(j,k,l);add(k,j,l);    }    tot=0;    fo(i,1,m){        ask[i][0]=read();ask[i][1]=read();        add2(ask[i][0],ask[i][1]);    }    build(1,0);    fo(i,1,n) mx[i]=se[i]=-inf;    fo(i,1,n) bz[i]=0;    solve(1,0);    printf("%lld\n",ans);}