GYM100524 H

题意：
给一棵n个点二叉树，要求支持删除叶节点同时维护轻重链剖分。m表示依次删除m个叶节点，删除每个节点后输出重孩子的标号和。
m< n<=200000

#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<iostream>
#define N 210000
#define LL long long
using namespace std;
struct node{int lc,rc,cnt;}lt[N*20];
int c[N][2],rt[N],tl,n,m,ddl[N],siz[N],tot[N],A[N],B[N];
LL s[N],ans;
void dfs(int x)
{
    siz[x]=1;
    if(c[x][0]) {dfs(c[x][0]);siz[x]+=siz[c[x][0]];}
    if(c[x][1]) {dfs(c[x][1]);siz[x]+=siz[c[x][1]];}
    if(siz[c[x][1]]>siz[c[x][0]]) swap(c[x][0],c[x][1]);
}
void merge(int &x,int y)
{
    if(y==0) return;
    if(x==0) {x=y;return;}
    lt[x].cnt+=lt[y].cnt;
    merge(lt[x].lc,lt[y].lc);
    merge(lt[x].rc,lt[y].rc);
}
void change(int &x,int k,int l,int r)
{
    if(x==0) {x=++tl;lt[x].lc=lt[x].rc=lt[x].cnt=0;}
    lt[x].cnt++;
    if(l==r) return;
    int mid=(l+r)/2;
    if(mid>=k) change(lt[x].lc,k,l,mid);
    else change(lt[x].rc,k,mid+1,r);
}
int find(int x,int k,int l,int r)
{
    if(lt[x].cnt<k) return m+1;
    if(l==r) return l;
    int lc=lt[x].lc,rc=lt[x].rc,mid=(l+r)/2;
    if(lt[lc].cnt>=k) return find(lc,k,l,mid);
    return find(rc,k-lt[lc].cnt,mid+1,r);
}
void solve(int x,int y)
{
    int num=siz[y],d=siz[x]-siz[y];
    for(int i=1;i<=num;i++)
    {
        A[i]=find(rt[x],i+d,1,m);
        B[i]=find(rt[y],i,1,m);
        if(A[i]==m+1 && B[i]==m+1) break;
    }
    num++;
    A[num]=B[num]=m+1;
    int i=1,j=1,pre=1,son=0,now,s0=siz[y],s1=siz[y];
    while(1)
    {
        int o;
        if(A[i]<B[j]) now=A[i++],o=0;
        else now=B[j++],o=1;
        if(son==0) {if(s0) s[pre]+=x,s[now]-=x;}
        else {if(s1) s[pre]+=y,s[now]-=y;}
        if(now==m+1) break;
        if(o==0)
        {
            s0--;
            if(son==0 && s0>=s1) son=0;
            else son=1;
        }
        else
        {
            s1--;
            if(son==1 && s1>=s0) son=1;
            else son=0;
        }
        pre=now;
    }
}
void dfs1(int x)
{
    int c0=c[x][0],c1=c[x][1];
    tot[x]=0;
    if(c0) {dfs1(c0);tot[x]+=tot[c0];}
    if(c1) {dfs1(c1);tot[x]+=tot[c1];}
    if(c0 && c1) solve(c0,c1);
    else if(c0) s[1]+=c0,s[ddl[c0]]-=c0;
    merge(rt[x],rt[c0]);
    merge(rt[x],rt[c1]);
    if(ddl[x]<=m) {change(rt[x],ddl[x],1,m);tot[x]++;}
}
int main()
{
    freopen("heavylight.in","r",stdin);
    freopen("heavylight.out","w",stdout);
    while(1)
    {
        scanf("%d",&n);
        if(n==0) break;
        for(int i=1;i<=n;i++) scanf("%d%d",&c[i][0],&c[i][1]);
        dfs(1);
        scanf("%d",&m);
        for(int i=1;i<=n;i++) ddl[i]=m+1;
        for(int i=1;i<=m;i++)
        {
            int x;scanf("%d",&x);
            ddl[x]=i;
        }
        for(int i=1;i<=n;i++) rt[i]=0;
        tl=0;
        for(int i=0;i<=m+1;i++) s[i]=0;
        dfs1(1);
        for(int i=1;i<=m;i++) s[i]+=s[i-1];
        ans=0;
        for(int i=1;i<=n;i++) ans+=c[i][0];
        printf("%lld\n",ans);
        for(int i=1;i<=m;i++) printf("%lld\n",s[i]);
    }
    return 0;
}

题解：
设点x两个孩子为y1,y2，初始重孩子是y1。注意x的重孩子的改变次数只有$O(siz[y2])$次，关键时间点只有y1的最后siz[y2]次删除和y2的删除。把关键时间点拿出来暴力，复杂度相当于对于每个节点遍历非重孩子子树复杂度，为$O(nlogn)$。
找关键点的实现只会log的，写了个线段树合并。总复杂度$O(nlog^2n)$。