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 long
using 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;
}