BZOJ4539: [Hnoi2016]树

WA到哭qwq
插入的点本身就在模板树上时不用找子树内第k大qwq

我们将每次插入的一棵子树看作一个块，那么在模板树上插入块，最后的大树会有n+m块
插入时如果接上去的父亲不在模板树上，二分找到接在哪一个块p上，再用主席树找p对应模板树的子树内第k大，从而找到接在块p对应模板树里的哪个节点fa上，算粗这个块的根到块p的根的距离作为这两个块之间的边权，顺便维护st表
询问时先在大树上跳块，直到跳到同一个块内，再在对应的模板树上跳

code：

#include<set>#include<map>#include<deque>#include<queue>#include<stack>#include<cmath>#include<ctime>#include<bitset>#include<string>#include<vector>#include<cstdio>#include<cstdlib>#include<cstring>#include<climits>#include<complex>#include<iostream>#include<algorithm>#define ll long longusing namespace std;const int maxn = 210000;const int maxd = 24;int n,m,q;struct edge{int y,nex;}a[maxn<<1]; int len,fir[maxn];inline void ins(const int x,const int y){a[++len]=(edge){y,fir[x]};fir[x]=len;}int siz[maxn],idt[maxn],dfn[maxn],ID,dep[maxn];int To[maxn],up[maxn];int fa[maxn][maxd];ll dis[maxn][maxd],sum[maxn];void dfs(const int x){    siz[x]=1; dfn[idt[++ID]=x]=ID;    for(int i=1;i<maxd;i++)         fa[x][i]=fa[fa[x][i-1]][i-1],        dis[x][i]=dis[x][i-1]+dis[fa[x][i-1]][i-1];    for(int k=fir[x],y=a[k].y;k;k=a[k].nex,y=a[k].y) if(y!=fa[x][0])    {        dep[y]=dep[x]+1; fa[y][0]=x; dis[y][0]=1;        dfs(y);        siz[x]+=siz[y];    }}int find_(const ll x){    int l=1,r=n;    while(l<=r)    {        int mid=l+r>>1;        if(sum[mid]>=x) r=mid-1;        else l=mid+1;    }    return r+1;}struct segment{int lc,rc,c;}seg[maxn*60]; int root[maxn],cnt;void merge(int &x,int y){    if(!y) return;    if(!x) { x=y;return; }    seg[x].c+=seg[y].c;    merge(seg[x].lc,seg[y].lc);    merge(seg[x].rc,seg[y].rc);}int loc;void upd(int &x,const int l,const int r){    if(!x) x=++cnt;    seg[x].c++;    if(l==r) return;    int mid=l+r>>1;    if(loc<=mid) upd(seg[x].lc,l,mid);    else upd(seg[x].rc,mid+1,r);}int query(const int x,const int y,const int l,const int r,int k){    if(l==r) return l;    int mid=l+r>>1,lc=seg[x].lc,ylc=seg[y].lc;    if(k>seg[ylc].c-seg[lc].c) return query(seg[x].rc,seg[y].rc,mid+1,r,k-(seg[ylc].c-seg[lc].c));    else return query(lc,ylc,l,mid,k);}int Query(const int l,const int r,const int k) { return query(root[l-1],root[r],1,ID,k); }int main(){    scanf("%d%d%d",&n,&m,&q);    for(int i=1;i<n;i++)    {        int x,y; scanf("%d%d",&x,&y);        ins(x,y); ins(y,x);    }    dep[1]=1; dfs(1);    for(int i=1;i<=ID;i++)    {        loc=idt[i]; upd(root[i],1,ID);        merge(root[i],root[i-1]);    }    for(int i=1;i<=n;i++) sum[i]=i,To[i]=i,up[i]=fa[i][0];    while(m--)    {        int x;ll y; scanf("%d%lld",&x,&y);        int id=find_(y);        n++; sum[n]=sum[n-1]+siz[x];        To[n]=x; up[n]=y<=ID?y:Query(dfn[To[id]],dfn[To[id]]+siz[To[id]]-1,y-sum[id-1]);        dep[n]=dep[id]+1;        fa[n][0]=id,dis[n][0]=dep[up[n]]-dep[To[id]]+1;        for(int i=1;i<maxd;i++)            fa[n][i]=fa[fa[n][i-1]][i-1],            dis[n][i]=dis[n][i-1]+dis[fa[n][i-1]][i-1];    }    while(q--)    {        ll x,y; scanf("%lld%lld",&x,&y);        int f1=find_(x),f2=find_(y); ll re=0;        x=x<=ID?x:Query(dfn[To[f1]],dfn[To[f1]]+siz[To[f1]]-1,x-sum[f1-1]);        y=y<=ID?y:Query(dfn[To[f2]],dfn[To[f2]]+siz[To[f2]]-1,y-sum[f2-1]);        if(f1!=f2)         {            if(dep[f1]<dep[f2]) swap(f1,f2),swap(x,y);            if(dep[f1]>dep[f2])            {                re+=dep[x]-dep[To[f1]]; x=To[f1];                for(int i=maxd-1;i>=0;i--)                    if(dep[fa[f1][i]]>dep[f2])                        re+=dis[f1][i],f1=fa[f1][i],x=To[f1];                if(fa[f1][0]!=f2) re+=dis[f1][0],f1=fa[f1][0],x=To[f1];                else re++,x=up[f1],f1=f2;            }            if(f1!=f2)            {                re+=dep[x]-dep[To[f1]]+dep[y]-dep[To[f2]];                x=To[f1],y=To[f2];                for(int i=maxd-1;i>=0;i--) if(fa[f1][i]!=fa[f2][i])                    re+=dis[f1][i],f1=fa[f1][i],x=To[f1],                    re+=dis[f2][i],f2=fa[f2][i],y=To[f2];                re++,x=up[f1];                re++,y=up[f2];            }        }        if(x!=y)        {            if(dep[x]<dep[y]) swap(x,y);            for(int i=maxd-1;i>=0;i--) if(dep[fa[x][i]]>=dep[y])                re+=(1<<i),x=fa[x][i];            for(int i=maxd-1;i>=0;i--) if(fa[x][i]!=fa[y][i])                re+=(1<<i+1),x=fa[x][i],y=fa[y][i];            if(x!=y) re+=2;        }        printf("%lld\n",re);    }    return 0;}