[JZOJ5131][SDOI省队集训2017]距离

题目描述

点分治

询问拆成四条到根的询问。
bfs一遍，处理出可持久化点分树。
查询直接查。

#include<cstdio>#include<algorithm>#include<cmath>#define fo(i,a,b) for(i=a;i<=b;i++)#define fd(i,a,b) for(i=a;i>=b;i--)using namespace std;typedef long long ll;const int maxn=400000+10,maxtot=maxn*30;int son[maxn],sx[maxn],fa[maxn];int rot[maxn],tree[maxtot][4];ll d1[maxn*20],d2[maxn*20];int s1[maxn*20],s2[maxn*20];int h[maxn],go[maxn*2],dis[maxn*2],next[maxn*2],p[maxn],bb[maxn],cc[maxn],zs[maxn];ll d[maxn][20];int belong[maxn][20],f[maxn][20],dep[maxn],zjy[maxn],sta[80];int size[maxn],a[maxn];int h2[maxn*2],g2[maxn*4],di2[maxn*4],n2[maxn*4];int ask[maxn];ll sum[maxn];bool bz[maxn];int i,j,k,l,r,t,nn,n,m,tot,top,cnt,num,czy;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;    next[tot]=h[x];    h[x]=tot;}void add2(int x,int y,int z){    g2[++num]=y;    di2[num]=z;    n2[num]=h2[x];    h2[x]=num;}void link(int x,int y,int z){    add2(x,y,z);    add2(y,x,z);}void new_build(int x,int y){    int t=h[x];    int i,j,k=0;    while (t){        if (go[t]!=y) bb[++k]=go[t],cc[k]=dis[t];        t=next[t];    }    if (k<=1){        fo(i,1,k) link(x,bb[i],cc[i]);    }    else{        fo(i,1,1) link(x,bb[i],cc[i]);        j=x;        fo(i,2,k){            ++n;            zs[n]=x;            link(j,n,0);            j=n;            link(j,bb[i],cc[i]);        }    }    t=h[x];    while (t){        if (go[t]!=y) new_build(go[t],x);        t=next[t];    }}void dfs(int x,int y){    f[x][0]=y;    dep[x]=dep[y]+1;    int t=h[x];    while (t){        if (go[t]!=y) dfs(go[t],x);        t=next[t];    }}int lca(int x,int y){    if (dep[x]<dep[y]) swap(x,y);    if (dep[x]!=dep[y]){        int j=zjy[dep[x]];        while (j>=0){            if (dep[f[x][j]]>=dep[y]) x=f[x][j];            j--;        }    }    if (x==y) return x;    int j=zjy[dep[x]];    while (j>=0){        if (f[x][j]!=f[y][j]){            x=f[x][j];            y=f[y][j];        }        j--;    }    return f[x][0];}void travel(int x,int y){    a[++top]=x;    int t=h[x];    size[x]=1;    while (t){        if (go[t]!=y&&!bz[go[t]]){            travel(go[t],x);            size[x]+=size[go[t]];        }        t=next[t];    }}void dg(int x,int y,int z,int w){    belong[x][w]=z;    int t=h[x];    while (t){        if (go[t]!=y&&!bz[go[t]]){            d[go[t]][w]=d[x][w]+(ll)dis[t];            dg(go[t],x,z,w);        }        t=next[t];    }}int solve(int x,int y){    top=0;    travel(x,0);    int i,j=x,k=0,t;    while (1){        t=h[j];        while (t){            if (go[t]!=k&&!bz[go[t]]&&size[go[t]]>top/2){                k=j;                j=go[t];                break;            }            t=next[t];        }        if (!t) break;    }    if (!y) rot[0]=j;    t=h[j];    while (t){        if (!bz[go[t]]){            //change(root[j],1,n,go[t],go[t]);            d[go[t]][y]=dis[t];            dg(go[t],j,j,y);        }        t=next[t];    }    bz[j]=1;    belong[j][y]=j;    t=h[j];    while (t){        if (!bz[go[t]]){            son[j]++;            k=solve(go[t],y+1);            sx[k]=son[j];            fa[k]=j;        }        t=next[t];    }    return j;}void bfs(int x,int y){    int i,j,z=p[x],w=rot[y],u,v;    if (x>nn){        rot[x]=rot[y];    }    else{    tot++;    rot[x]=tot;    fo(i,0,20){        //root[tot]=root[w];        fo(j,1,3) tree[tot][j]=tree[w][j];        d1[tot]=d1[w]+d[z][i];        s1[tot]=s1[w]+1;        if (i){            d2[tot]=d2[w]+d[z][i-1];            s2[tot]=s2[w]+1;        }        if (belong[z][i]==z) break;        u=sx[belong[z][i+1]];        v=son[belong[z][i]];        //change(root[tot],1,v,u,tot+1);        tree[tot][u]=tot+1;        tot++;        //w=query(root[w],1,v,u);        w=tree[w][u];    }    }    int t=h[x];    while (t){        if (go[t]!=y) bfs(go[t],x);        t=next[t];    }}void calc(int x,int f){    if (!x) return;    int i,y=rot[x];    fo(i,0,20){        if (i){            ans-=f*d2[y];            ans-=(ll)f*s2[y]*d[k][i-1];        }        ans+=f*d1[y];        ans+=(ll)f*s1[y]*d[k][i];        if (belong[k][i]==k) break;        //y=query(root[y],1,son[belong[k][i]],sx[belong[k][i+1]]);        y=tree[y][sx[belong[k][i+1]]];    }}void write(ll x){    if (!x){        putchar('0');        putchar('\n');        return;    }    top=0;    while (x){        sta[++top]=x%10;        x/=10;    }    while (top) putchar(sta[top--]+'0');    putchar('\n');}int main(){    freopen("dis.in","r",stdin);freopen("dis.out","w",stdout);    czy=read();    n=read();m=read();    nn=n;    fo(i,1,n-1){        j=read();k=read();l=read();        add(j,k,l);add(k,j,l);    }    new_build(1,0);    fo(i,1,n) h[i]=h2[i];    fo(i,1,num){        go[i]=g2[i];        dis[i]=di2[i];        next[i]=n2[i];    }    dfs(1,0);    fo(i,1,n) zjy[i]=floor(log(i)/log(2));    fo(j,1,zjy[n])        fo(i,1,n)            f[i][j]=f[f[i][j-1]][j-1];    fo(i,1,nn) p[i]=read();    tot=top=cnt=0;    solve(1,0);    fo(i,1,n)        if (sx[i]) tree[fa[i]][sx[i]]=i;    tot=n;    fo(i,1,nn) zs[i]=i;    bfs(1,0);    while (m--){        l=read();r=read();k=read();        l^=(czy*ans);r^=(czy*ans);k^=(czy*ans);        ans=0;        calc(l,1);        calc(r,1);        j=zs[lca(l,r)];        calc(j,-1);        j=zs[f[j][0]];        calc(j,-1);        write(ans);    }}