BZOJ3223&3224 文艺&普通平衡树 Splay模板(数组)

13 篇文章 0 订阅
3 篇文章 0 订阅
才发现自己splay的模板又臭又长,还自带大常数,而且指针不会调,换回数组,从新打了一遍

区间翻转:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int read()
{
    int x=0,f=1; char ch=getchar();
    while (ch<'0' || ch>'9') {if (ch=='-') f=-1; ch=getchar();}
    while (ch>='0' && ch<='9') {x=x*10+ch-'0'; ch=getchar();}
    return x*f;
}
#define maxn 101000
#define inf 1000000000

int son[maxn][2],fa[maxn],key[maxn],size[maxn],rev[maxn],sz,root;
int n,m,a[maxn]={0};

void update(int x)
{
    size[x]=size[son[x][1]]+size[son[x][0]]+1;
}
void swap(int &x,int &y)
{
    int tmp=x;x=y;y=tmp;
}
void pushdown(int x)
{
    if (!x || !rev[x]) return;
    rev[son[x][0]]^=1; rev[son[x][1]]^=1;
    swap(son[x][0],son[x][1]);
    rev[x]=0;
}
int build(int left,int right,int f)
{
    if (left>right) return 0;
    int mid=(left+right)>>1;
    int now=++sz;
    key[now]=a[mid]; fa[now]=f; rev[now]=0;
    int leftchild=build(left,mid-1,now);
    int rightchild=build(mid+1,right,now);
    son[now][0]=leftchild; son[now][1]=rightchild; update(now);
    return now;
}
int get(int x)
{
    return son[fa[x]][1]==x;
}
void rotate(int x)
{
    pushdown(fa[x]); pushdown(x);
    int old=fa[x],oldf=fa[old],which=get(x);
    son[old][which]=son[x][which^1]; fa[son[old][which]]=old;
    fa[old]=x; son[x][which^1]=old; fa[x]=oldf;
    if (oldf)
        son[oldf][son[oldf][1]==old]=x;
    update(old); update(x);
}
void splay(int x,int tar)
{
    for (int f; (f=fa[x])!=tar; rotate(x))
        if (fa[f]!=tar)
            rotate(get(x)==get(f)?f:x);
    if (tar==0) root=x;
}
int find(int x)
{
    int now=root;
    while (true)
    {
        pushdown(now);
        if (x<=size[son[now][0]])
            now=son[now][0];
        else
        {
            x-=size[son[now][0]]+1;
            if (x==0) return now;
            now=son[now][1];
        }
    }
    return -1;
}
void print(int now)
{
    pushdown(now);
    if (son[now][0]) print(son[now][0]);
    if (key[now]!=inf && key[now]!=-inf)
        printf("%d ",key[now]);
    if (son[now][1]) print(son[now][1]);
}

int main()
{
    n=read(),m=read();
    a[1]=-inf; a[n+2]=inf;
    for (int i=1; i<=n; i++) a[i+1]=i;
    root=build(1,n+2,0);
    for (int i=1;i<=m;i++)
    {
        int x=read(),y=read();
        if (x>=y) continue;
        int aa=find(x); int bb=find(y+2);
        splay(aa,0); splay(bb,aa);
        rev[son[son[root][1]][0]]^=1;
    }
    print(root);
    printf("\n");  
    return 0;
}

插入,删除,前驱,后继,编号为k的值,排名为k的值:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
int read()
{
    int x=0,f=1; char ch=getchar();
    while (ch<'0' || ch>'9') {if (ch=='-') f=-1; ch=getchar();}
    while (ch>='0' && ch<='9') {x=x*10+ch-'0'; ch=getchar();}
    return x*f;
}
#define maxn 1000010
int n,m;
int fa[maxn],son[maxn][2],key[maxn],cnt[maxn],size[maxn];
int sz,root;
    void clear(int now)
        {
            son[now][1]=son[now][0]=fa[now]=cnt[now]=key[now]=size[now];
        }
    int get(int now)
        {
            return son[fa[now]][1]==now;
        }
    void update(int now)
        {
            if (!now) return;
            size[now]=cnt[now];
            if (son[now][0]) size[now]+=size[son[now][0]];
            if (son[now][1]) size[now]+=size[son[now][1]];
        }
    void rotate(int now)
        {
            int old=fa[now],oldf=fa[old],which=get(now);
            son[old][which]=son[now][which^1]; fa[son[old][which]]=old;
            fa[old]=now; son[now][which^1]=old; fa[now]=oldf;
            if (oldf) son[oldf][son[oldf][1]==old]=now;
            update(old); update(now);
        }
    void splay(int now)
        {
            for (int f; (f=fa[now]); rotate(now))
                if (fa[f]) 
                    if (get(now)==get(f)) 
                        rotate(f); else rotate(now);
            root=now;
        }
    void insert(int v)
        {
            if (!root) 
                {sz++;son[sz][0]=son[sz][1]=fa[sz]=0;key[sz]=v;size[sz]=1;cnt[sz]=1;root=sz;return;}
            int now=root,f=0;
            while (true)
                {
                    if (key[now]==v)
                        {cnt[now]++;update(now);update(f);splay(now);break;}
                    f=now; now=son[now][key[now]<v];
                    if (now==0)
                        {
                           sz++;son[sz][0]=son[sz][1]=0;key[sz]=v;size[sz]=1;
                           cnt[sz]=1;fa[sz]=f;son[f][key[f]<v]=sz;
                           update(f);splay(sz);break;
                        }
                }
        }
    int find(int v)
        {
            int ans=0,now=root;
            while (true)
                {
                    if (v<key[now]) now=son[now][0]; else
                        {
                            if (son[now][0]) ans+=size[son[now][0]];
                            if (v==key[now]) {splay(now); return ans+1;}
                            ans+=cnt[now]; now=son[now][1];
                        }
                }
        }
    int findkth(int x)
        {
            int now=root;
            while (true)
                {
                    if (son[now][0] && x<=size[son[now][0]]) now=son[now][0]; else
                        {
                            int temp;
                            if (son[now][0]) temp=size[son[now][0]]+cnt[now];
                                        else temp=cnt[now];
                            if (x<=temp) return key[now];
                            x-=temp; now=son[now][1];
                        }
                }
        }
    int prev()
        {
            int now=son[root][0];
            while (son[now][1]) now=son[now][1];
            return now;
        }
    int succ()
        {
            int now=son[root][1];
            while (son[now][0]) now=son[now][0];
            return now;
        }
    bool remove(int now)
        {
            if (find(now)==-1) return false;
            if (cnt[root]>1) {cnt[root]--; return true;};
            if (!son[root][0] && !son[root][1]) {clear(root); root=0; return true;}
            if (!son[root][0])
                {
                    int oldroot=root; root=son[root][1]; fa[root]=0; clear(oldroot); return true;
                }
            else if (!son[root][1])
                {
                    int oldroot=root; root=son[root][0]; fa[root]=0; clear(oldroot); return true;
                }
            int leftbig=prev(),oldroot=root;
            splay(leftbig);
            fa[son[oldroot][1]]=root; son[root][1]=son[oldroot][1];
            clear(oldroot); update(root); return true;
        }

int main()
{
    n=read();
    for (int i=1; i<=n; i++)
        {
            int opt=read(),x=read();
            switch (opt)
                {
                    case 1: insert(x);break;
                    case 2: remove(x);break;
                    case 3: printf("%d\n",find(x));break;
                    case 4: printf("%d\n",findkth(x));break;
                    case 5: insert(x); printf("%d\n",key[prev()]); remove(x);break;
                    case 6: insert(x); printf("%d\n",key[succ()]); remove(x);break;
                }
        }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值