BZOJ 3720 Gty的妹子树

树分块

每S个节点分一块，如果x的父亲所属块超过S就自己分一块。
块内的所有权值排序。
对于询问，块内二分，块外暴力。
对于修改和加入，维护排序就好。

#include<cstdio>
#include<vector>
#include<algorithm>
#define N 60005
#define S 472//经过计算 这个值是最佳的 
using namespace std;
struct edge{int next,to;}e[N*10];
int w[N], last[N], last2[N], ecnt=1, belong[N], bcnt, top[N], siz[N], q[N], fa[N], n;
vector<int> data[N];
void add1(int a, int b)
{
    e[++ecnt]=(edge){last[a],b};
    last[a]=ecnt;
}
void add2(int a, int b)
{
    e[++ecnt]=(edge){last2[a],b};
    last2[a]=ecnt;
}
void newnode(int x, int f)
{
    if(siz[belong[f]]<S)
    {
        belong[x]=belong[f];
    }
    else
    {
        belong[x]=++bcnt;
        top[belong[x]]=x;
        add2(top[belong[f]],x);
    }
    ++siz[belong[x]];
    data[belong[x]].push_back(w[x]);
}
void bfs()
{
    belong[0]=bcnt=1;
    top[bcnt]=1;
    q[0]=1;
    for(int head=0, tail=1; head<tail; head++)
    {
        int x=q[head], f=fa[x];
        newnode(x,f);
        for(int i = last[x]; i; i = e[i].next)
        {
            int y=e[i].to;
            if(belong[y])continue;
            fa[y]=x;
            q[tail++]=y;
        }
    }
}
int solve(int x, int val)
{
    int l=-1, r=siz[x]-1;
    while(l<r)
    {
        int mid=(l+r+1)>>1;
        if(data[x][mid]<=val)l=mid;
        else r=mid-1;
    }
    return siz[x]-l-1;
}
int query(int x, int val)
{
    int ret=0;
    if(top[belong[x]]==x)
    {
        ret+=solve(belong[x], val);
        for(int i = last2[x]; i; i=e[i].next)
            ret+=query(e[i].to,val);
    }
    else
    {
        ret+=(w[x]>val);
        for(int i = last[x]; i; i=e[i].next)
        {
            if(fa[e[i].to]==x)
                ret+=query(e[i].to,val);
        }
    }
    return ret;
}
void modify(int x, int val)
{
    int block=belong[x], pos;
    for(pos = 0; pos < siz[block]; pos++)
        if(data[block][pos]==w[x])break;
    data[block][pos]=w[x]=val;
    while(pos < siz[block]-1 && data[block][pos]>data[block][pos+1])
    {
        swap(data[block][pos], data[block][pos+1]);
        pos++;
    }
    while(pos > 0 && data[block][pos]<data[block][pos-1])
    {
        swap(data[block][pos], data[block][pos-1]);
        pos--;
    }
}
void add(int f, int val)
{
    w[++n]=val;
    add1(f,n);
    fa[n]=f;
    newnode(n,f);
    for(int pos=siz[belong[n]]-1; pos>0; pos--)
    {
        if(data[belong[n]][pos]<data[belong[n]][pos-1])
            swap(data[belong[n]][pos],data[belong[n]][pos-1]);
        else break;
    }
}
int main()
{
    scanf("%d",&n);
    for(int i = 1, u, v; i < n; i++)
    {
        scanf("%d%d",&u,&v);
        add1(u,v);
        add1(v,u);
    }
    for(int i = 1; i <= n; i++)
        scanf("%d",&w[i]);
    bfs();
    for(int i = 1; i <= bcnt; i++)
        sort(data[i].begin(), data[i].end());
    int m;
    scanf("%d", &m);
    int ans=0;
    while(m--)
    {
        int op, u, v;
        scanf("%d%d%d", &op, &u, &v);
        u^=ans; v^=ans; 
        if(op==0)
        {
            ans=query(u,v);
            printf("%d\n",ans);
        }
        else if(op==1)
            modify(u,v);
        else
            add(u,v);
    }
}