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

来源:互联网 发布:汉字听写大赛软件 编辑:程序博客网 时间:2024/04/27 15:12

把树分成大小为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 longusing 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;}
0 0