BZOJ 3720: Gty的妹子树|树分块

把树分成大小为sqrt(n)的块
如果x的fa[x]所属的块大小等于根号n了，那么久新开一块否则就属于fa[x]的块
每一块内把数排序，询问时块内二分查找，块外暴力

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<iostream>
#include<algorithm>
#define lowbit(x) (x&(-x))
#define ll long long
using namespace std;
int sc()
{
    int i=0;char c=getchar();
    while(c>'9'||c<'0')c=getchar();
    while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();
    return i;
}
struct BLOCK{
    int a[188],size;
    void insert(int x)
    {
        a[++size]=x;
    }
    void change(int x,int y)
    {
        int t=upper_bound(a+1,a+size+1,x)-a-1;
        a[t]=y; 
        sort(a+1,a+size+1);
    }
    int query(int x)
    {
        int t=upper_bound(a+1,a+size+1,x)-a;
        return size-t+1;
    }
}block[10001];
int head[60030],lst[90030],nxt[90030];
int Head[30030],Lst[60030],Nxt[60030];
int a[60030],fa[60030],bl[60030];
int m,n,Block,cnt,ans=0,tot,Tot;
void insert(int x,int y)
{
    lst[++tot]=y;nxt[tot]=head[x];head[x]=tot;
}
void Insert(int x,int y)
{
    Lst[++Tot]=y;Nxt[Tot]=Head[x];Head[x]=Tot;
}
void dfs(int x)
{
    if(block[bl[fa[x]]].size==Block)
        block[bl[x]=++cnt].insert(a[x]),Insert(bl[fa[x]],cnt);
    else 
        block[bl[x]=bl[fa[x]]].insert(a[x]);
    for(int i=head[x];i;i=nxt[i])
        if(lst[i]!=fa[x])
            fa[lst[i]]=x,dfs(lst[i]);
}
void Block_DFS(int x,int y)
{
    ans+=block[x].query(y);
    for(int i=Head[x];i;i=Nxt[i])
        Block_DFS(Lst[i],y);
}
void DFS(int x,int y)
{
    //cout << x <<" "<< ans<<endl;
    if(a[x]>y)ans++;
    for(int i=head[x];i;i=nxt[i])
        if(lst[i]!=fa[x])
        {
            if(bl[lst[i]]==bl[x]) DFS(lst[i],y);
            else  Block_DFS(bl[lst[i]],y);
        }
}
int main()
{
    n=sc();Block=sqrt(n);
    for(int i=1;i<n;i++)
    {
        int x=sc(),y=sc();
        insert(x,y);
        insert(y,x);
    }
    for(int i=1;i<=n;i++)a[i]=sc();
    dfs(1);
    for(int i=1;i<=cnt;i++)
        sort(block[i].a+1,block[i].a+block[i].size+1);
    m=sc();
    for(int i=1;i<=m;i++)
    {
        int opt=sc(),x=sc(),y=sc();
        x^=ans,y^=ans;
        if(opt==0)
        {
            ans=0;
            DFS(x,y);
            printf("%d\n",ans);
        }
        else if(opt==1)
        {
            block[bl[x]].change(a[x],y);
            a[x]=y;
        }
        else
        {
            a[++n]=y;
            insert(x,n);
            fa[n]=x;
            if(block[bl[x]].size==Block)
                block[bl[n]=++cnt].insert(y),Insert(bl[x],cnt);
            else
            {
                block[bl[n]=bl[x]].insert(y);
                sort(block[bl[n]].a+1,block[bl[n]].a+block[bl[n]].size+1);
            }
        }
    }
    return 0;
}