3720: Gty的妹子树 块状树

将树分块,保证每块内都联通,然后保证每块内都有序,就可以支持查询操作了。
加入节点的时候,如果达到块的大小就重建一个块,否则加入当前块,不要忘记加入后对块排序!

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
const int N=60005;
int head[N],list[N<<1],next[N<<1];
int HEAD[N],LIST[N<<1],NEXT[N<<1];
int a[N],fa[N],belong[N];
int n,m,cnt,ans,CNT,tot,B;
struct node
{
    int size;
    int a[205];
    inline void insert(int x)
    {
        a[++size]=x;
    }
    inline void change(int x,int y)
    {
        int t=lower_bound(a+1,a+size+1,x)-a;
        a[t]=y;
        sort(a+1,a+size+1);
    }
    inline int query(int x)
    {
        int t=upper_bound(a+1,a+size+1,x)-a;
        return size-t+1;
    }
};
node block[10005];
inline int read()
{
    int a=0,f=1; char c=getchar();
    while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}
    while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}
    return a*f;
}
inline void insert(int x,int y)
{
    next[++cnt]=head[x];
    head[x]=cnt;
    list[cnt]=y;
}
inline void INSERT(int x,int y)
{
    NEXT[++CNT]=HEAD[x];
    HEAD[x]=CNT;
    LIST[CNT]=y;
}
void dfs(int x)
{
    if (block[belong[fa[x]]].size==B)
        belong[x]=++tot,block[tot].insert(a[x]),INSERT(belong[fa[x]],tot);
    else belong[x]=belong[fa[x]],block[belong[x]].insert(a[x]);
    for (int i=head[x];i;i=next[i])
        if (list[i]!=fa[x])
            fa[list[i]]=x,dfs(list[i]);
}
void dfs1(int x,int y)
{
    ans+=block[x].query(y);
    for (int i=HEAD[x];i;i=NEXT[i])
        dfs1(LIST[i],y);
}
void query(int x,int y)
{
    if (a[x]>y) ans++;
    for (int i=head[x];i;i=next[i])
        if (list[i]!=fa[x])
        {
            if (belong[list[i]]==belong[x]) query(list[i],y);
            else dfs1(belong[list[i]],y);
        }
}
int main()
{
    n=read(); B=sqrt(n); 
    for (int i=1;i<n;i++)
    {
        int u=read(),v=read();
        insert(u,v); insert(v,u);
    }
    for (int i=1;i<=n;i++) a[i]=read();
    dfs(1);
    for (int i=0;i<=tot;i++)
        sort(block[i].a+1,block[i].a+block[i].size+1);
    m=read();
    for (int i=1;i<=m;i++)
    {
        int opt=read(),u=read()^ans,x=read()^ans;
        if (opt==0)
        {
            ans=0;
            query(u,x);
            printf("%d\n",ans);
        }
        else if (opt==1)
        {
            block[belong[u]].change(a[u],x);
            a[u]=x;
        }
        else
        {
            a[++n]=x;
            insert(u,n);
            fa[n]=u;
            if (block[belong[u]].size==B)
                belong[n]=++tot,block[tot].insert(x),INSERT(belong[u],tot);
            else
            {
                belong[n]=belong[u],block[belong[n]].insert(a[n]);
                sort(block[belong[n]].a+1,block[belong[n]].a+block[belong[n]].size+1);
            }
        }
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值