bzoj 3678 wangxz与OJ

3678: wangxz与OJ

Time Limit: 10 Sec  Memory Limit: 128 MB
http://www.lydsy.com/JudgeOnline/problem.php?id=3678

Description

 

某天,wangxz神犇来到了一个信息学在线评测系统(Online Judge)。由于他是一位哲♂学的神犇,所以他不打算做题。他发现这些题

目呈线性排列,被标记为1~n号,每道题都有一个难度值(可以<=0)。他决定与这些题目玩♂耍。

1、他可以在某个位置插♂入一些难度值特定的题目。

2、他可以吃♂掉(删除)一段题目。

3、他可以查询某个位置的题目的难度值。

维护一个初始有n个元素的序列(标记为1~n号元素),支持以下操作:

0 p a b (0<=p<=当前序列元素个数) (a<=b) 在p位置和p+1位置之间插入整数:a,a+1,a+2,...,b-1,b。若p为0,插在序列最前面;

1 a b (1<=a<=b<=当前序列元素个数) 删除a,a+1,a+2,...,b-1,b位置的元素;

2 p (1<=p<=当前序列元素个数) 查询p位置的元素。

 

Input

输入第一行包括两个正整数n(1<=n<=20000),m(1<=m<=20000),代表初始序列元素个数和操作个数。

接下来n个整数,为初始序列元素。

接下来m行,每行第一个为整数sym,

若sym=0,接下来有一个非负整数p,两个整数a,b;

若sym=1,接下来有两个正整数a,b;

若sym=2,接下来有一个正整数p;

p、x、y的含义及范围见题目描述。

在任何情况下,保证序列中的元素总数不超过100000。

保证题目涉及的所有数在int内。

Output

对每个sym=2,输出一行,包括一个整数,代表询问位置的元素。

Sample Input

5 3
1 2 3 4 5
0 2 1 4
1 3 8
2 2

Sample Output

2
 
插入时插一个大节点,使用到他时分裂节点
大节点发现是公差为1的等差数列,所以记录首项、末项即可
#include<cstdio>
#define N 20001
using namespace std;
int n,m,tot,t1,t2,root;
int key[N*10][2],siz[N*10],ch[N*10][2],fa[N*10];
int read()
{
    int x=0,f=1; char c=getchar();
    while(c<'0'||c>'9') { if(c=='-') f=-1; c=getchar(); }
    while(c>='0'&&c<='9') { x=x*10+c-'0'; c=getchar(); }
    return x*f;
}
struct SPLAY_TREE
{
    void update(int k)
    {
        siz[k]=key[k][1]-key[k][0]+1;
        if(ch[k][0]) siz[k]+=siz[ch[k][0]];
        if(ch[k][1]) siz[k]+=siz[ch[k][1]];
    }
    void build(int l,int r,int f)
    {
        if(l>r) return;
        int mid=l+r>>1;
        siz[mid]=1; fa[mid]=f; ch[f][mid>f]=mid;
        build(l,mid-1,mid);
        build(mid+1,r,mid);
        update(mid);
    }
    void rotate(int x,int & goal)
    {
        int y=fa[x],z=fa[y],k=ch[y][1]==x;
        if(y==goal) goal=x;
        else ch[z][ch[z][1]==y]=x;
        ch[y][k]=ch[x][k^1]; ch[x][k^1]=y;
        fa[y]=x; fa[ch[y][k]]=y; fa[x]=z;
        update(y);
    }
    bool getson(int x)
    {
        return ch[fa[x]][1]==x;
    }
    void splay(int x,int & goal)
    {
        while(x!=goal)
        {
            int y=fa[x];
            if(y!=goal) 
            {
                if(getson(x)==getson(y)) rotate(y,goal);
                else rotate(x,goal);
            }
            rotate(x,goal);
            update(x);
        }
    }
    int find(int now,int k)
    {
        if(k<=siz[ch[now][0]]) return find(ch[now][0],k);
        else if(siz[now]-siz[ch[now][1]]<k) return find(ch[now][1],k-siz[now]+siz[ch[now][1]]);
        else
        {
            k-=siz[ch[now][0]];
            if(k!=1)
            {
                fa[ch[++tot][0]=ch[now][0]]=tot; fa[ch[now][0]=tot]=now;
                key[tot][0]=key[now][0]; key[tot][1]=key[tot][0]+k-2; key[now][0]=key[tot][1]+1; 
                update(tot);
                k=1;
            }
            if(k!=key[now][1]-key[now][0]+1)
            {
                fa[ch[++tot][1]=ch[now][1]]=tot; fa[ch[now][1]=tot]=now;
                key[tot][1]=key[now][1]; key[tot][0]=key[now][0]+k; key[now][1]=key[tot][0]-1;
                update(tot);
            }
            return now;
        } 
    }
    void spilt(int l,int r)
    {
        t1=find(root,l);
        t2=find(root,r);
        splay(t1,root);
        splay(t2,ch[root][1]);
    }
}Splay;
int main()
{
    /*freopen("data.txt","r",stdin);
    freopen("my2.txt","w",stdout);*/
    n=read(); m=read();
    for(int i=1;i<=n;i++) key[i+1][0]=key[i+1][1]=read();
    Splay.build(1,n+2,0);
    tot=n+2; root=n+3>>1;
    int op,p,a,b;
    while(m--)
    {
        scanf("%d",&op);
        if(!op)
        {
            p=read(); a=read(); b=read();
            Splay.spilt(p+1,p+2);
            fa[ch[t2][0]=++tot]=t2; key[tot][0]=a; key[tot][1]=b; siz[tot]=b-a+1;
            Splay.update(t2); Splay.update(t1); 
        }
        else if(op==1)
        {
            a=read(); b=read();
            Splay.spilt(a,b+2);
            ch[t2][0]=0;
            Splay.update(t2); Splay.update(t1);
        }
        else
        {
            p=read();
            p=Splay.find(root,p+1);
            printf("%d\n",key[p][0]);
        }
    }
}

狂WA代码:

#include<cstdio>
#define N 20005
using namespace std;
int n,m,now,tmp,root;
int a[N];
int siz[N*200],fa[N*200],key[N*200],ch[N*200][2],cnt[N*200];
int tot;
struct SPLAY_TREE
{
    void update(int k)
    {
        siz[k]=cnt[k];
        if(ch[k][0]) siz[k]+=siz[ch[k][0]];
        if(ch[k][1]) siz[k]+=siz[ch[k][1]];
    }
    void build(int l,int r,int f)
    {
        if(l>r) return;
        int mid=l+r>>1;
        if(mid<f) ch[f][0]=mid;
        else  ch[f][1]=mid;
        siz[mid]=1; cnt[mid]=1; fa[mid]=f; key[mid]=a[mid];
        build(l,mid-1,mid);
        build(mid+1,r,mid);
        update(mid);
    }
    int find_id(int x)
    {
        now=root;
        while(1)
        {
            if(ch[now][0]&&siz[ch[now][0]]>=x) now=ch[now][0];
            else
            {
                tmp=(ch[now][0] ? siz[ch[now][0]] : 0)+cnt[now];
                if(x<=tmp)  return now;
                x-=tmp;
                now=ch[now][1];
            }
        }
    }
    int getson(int x)
    {
        return ch[fa[x]][0]==x;
    }
    void rotate(int x,int & goal)
    {
        int y=fa[x],z=fa[y],k=ch[y][1]==x;
        if(y==goal) goal=x;
        else ch[z][ch[z][1]==y]=x;
        ch[y][k]=ch[x][k^1]; ch[x][k^1]=y;
        fa[y]=x; fa[ch[y][k]]=y; fa[x]=z;
        update(y);
    }
    void splay(int x,int & goal)
    {
        int y;
        while(x!=goal)
        {
            y=fa[x];
            if(y!=goal)
            {
                if(getson(x)==getson(y)) rotate(y,goal);
                else rotate(x,goal);
            }
            rotate(x,goal);
            update(x);
        }
    }
    void insert_suf(int s,int t)
    {
        if(!ch[root][1]) now=root;
        else
        {
            now=ch[root][1];
            while(ch[now][0]) now=ch[now][0];
        }   
        siz[++tot]=cnt[tot]=t-s+1;
        key[tot]=(s+t)*(t-s+1)/2;
        fa[tot]=now;
        ch[now][0]=tot;
        splay(tot,root);
    }
    void insert_pre(int s,int t)
    {
        if(!ch[root][0]) now=root;
        else
        {
            now=ch[root][0];
            while(ch[now][1]) now=ch[now][1];
        }   
        siz[++tot]=cnt[tot]=t-s+1;
        key[tot]=(s+t)*(t-s+1)/2;
        fa[tot]=now;
        ch[now][1]=tot;
        splay(tot,root);
    }   
    int find_pre()
    {
        now=ch[root][0];
        while(ch[now][1]) now=ch[now][1];
        return now; 
    }
    int find_suf()
    {
        now=ch[now][1];
        while(ch[now][0]) now=ch[now][0];
        return now;
    }
    void del()
    {
        if(!ch[root][0]&&!ch[root][1])  root=0;
        else if(!ch[root][0])
        {
            tmp=root;
            root=ch[root][1];
            fa[root]=0;
            ch[tmp][1]=0;
        }
        else if(!ch[root][1])
        {
            tmp=root;
            root=ch[root][0];
            fa[root]=0;
            ch[tmp][0]=0;
        }
        else
        {
            int pre=find_pre();
            tmp=root;
            splay(pre,root);
            ch[root][1]=ch[tmp][1];
            if(ch[root][1]) fa[ch[root][1]]=root;
            update(root);
        }
    }
}Splay;
int main()
{
    /*freopen("data.txt","r",stdin);
    freopen("my.txt","w",stdout);*/
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&a[i+1]);
    Splay.build(1,n+2,0);
    root=n+3>>1;
    int op,p,aa,bb,q;
    int cutl,cutr;
    int shouxiang1,shouxiang2,moxiang1,moxiang2,rest_in_aa,rest_in_bb;
    tot=n+3;
    while(m--)
    {
        scanf("%d",&op);
        if(op==0)
        {
            scanf("%d%d%d",&p,&aa,&bb);
            p++;q=p; 
            p=Splay.find_id(p);
            Splay.splay(p,root);
            if(cnt[p]==1) Splay.insert_suf(aa,bb); 
            else
            {
                cutl=q-siz[ch[root][0]]-1;
                cutr=siz[root]-siz[ch[root][1]]-q;
                shouxiang1=(2*key[root]/cnt[root]-cnt[root]+1)/2;
                moxiang1=shouxiang1+cutl-1;
                shouxiang2=moxiang1+2;
                moxiang2=shouxiang2+cutr-1;
                siz[root]=siz[root]-cnt[root]+1;
                cnt[root]=1;
                key[root]=moxiang1+1;
                Splay.insert_pre(shouxiang1,moxiang1);
                Splay.splay(p,root);
                Splay.insert_suf(shouxiang2,moxiang2);
                Splay.splay(p,root);
                Splay.insert_suf(aa,bb);
            }
        }
        else if(op==1)
        {
            scanf("%d%d",&aa,&bb);
            aa++; bb++;
            p=aa; q=bb;
            aa=Splay.find_id(aa);
            bb=Splay.find_id(bb);
            if(aa==bb)
            {
                Splay.splay(aa,root);
                cutl=p-siz[ch[root][0]]-1;
                //cutr=siz[root]-siz[ch[root][1]]-q;
                cutr=cnt[root]-cutl-(q-p+1);
                shouxiang1=(2*key[root]/cnt[root]-cnt[root]+1)/2;
                moxiang1=shouxiang1+cutl-1;
                shouxiang2=moxiang1+q-p+2;//+1
                moxiang2=shouxiang2+cutr-1;
                if(shouxiang1<=moxiang1) 
                {
                    Splay.insert_pre(shouxiang1,moxiang1);//++
                    Splay.splay(aa,root);
                    siz[root]=siz[root]-(moxiang1-shouxiang1+1);
                    cnt[root]=cnt[root]-(moxiang1-shouxiang1+1);
                    key[root]=key[root]-(shouxiang1+moxiang1)*(moxiang1-shouxiang1+1)/2;
                }
                siz[root]=siz[root]-(q-p+1);
                cnt[root]=cnt[root]-(q-p+1);
                key[root]=key[root]-(shouxiang2-1+shouxiang2-1-(q-p+1)+1)*(q-p+1)/2;
                if(!cnt[root]) Splay.del();
                continue;
            }
            Splay.splay(aa,root);
            rest_in_aa=p-siz[ch[root][0]]-1;
            Splay.splay(bb,ch[root][1]);
            rest_in_bb=cnt[bb]-(q-p+1-(cnt[aa]-rest_in_aa+siz[ch[bb][0]]));//+1
            siz[aa]-=siz[ch[bb][0]];
            siz[bb]-=siz[ch[bb][0]];
            ch[bb][0]=0;
            shouxiang1=(2*key[bb]/cnt[bb]-cnt[bb]+1)/2;
            key[bb]-=(shouxiang1+shouxiang1+cnt[bb]-rest_in_bb-1)*(cnt[bb]-rest_in_bb)/2;
            cnt[bb]=rest_in_bb;
            Splay.update(bb);
            shouxiang1=(2*key[aa]/cnt[aa]-cnt[aa]+1)/2;
            key[aa]=(shouxiang1+shouxiang1+rest_in_aa-1)*rest_in_aa/2;
            cnt[aa]=rest_in_aa;
            Splay.update(aa);
        }
        else
        {
            scanf("%d",&p);
            p++;
            q=Splay.find_id(p);
            if(cnt[q]==1) printf("%d\n",key[q]);
            else
            {
                Splay.splay(q,root);
                p=p-siz[ch[root][0]];
                shouxiang1=(2*key[root]/cnt[root]-cnt[root]+1)/2;
                printf("%d\n",shouxiang1+p-1);
            }
        }
    }
}
View Code

 

作者:xxy
本文版权归作者和博客园共有,转载请用链接,请勿原文转载,Thanks♪(・ω・)ノ。
posted @ 2017-05-10 10:02  TRTTG  阅读(379)  评论(0编辑  收藏  举报