zoj 3612 Median(Treap)

来源：互联网发布：中国植物信息数据库编辑：程序博客网时间：2024/05/01 13:43

题意：给出两种操作，一种是在集合中添加一个数，另一种是删除一个数，每次操作如果合法，那么输出中位数。

思路：想到用Treap搞就直接写了，貌似方法还挺多的？每次在Treap中查询第k大就行了。对于偶数要查两个数，还是挺好写的吧。。

代码：

#include<iostream>#include<cstdio>#include<cstring>#include<string>#include<algorithm>#include<map>#include<queue>#include<stack>#include<set>#include<cmath>#include<vector>#define inf 0x3f3f3f3f#define Inf 0x3FFFFFFFFFFFFFFFLL#define eps 1e-9#define pi acos(-1.0)using namespace std;typedef long long ll;const int maxn=10000+10;int ch[maxn][2],val[maxn],counts[maxn],r[maxn],size[maxn],tot,root;void NewNode(int &rt,int v){    rt=++tot;    ch[rt][0]=ch[rt][1]=0;    val[rt]=v;    counts[rt]=size[rt]=1;    r[rt]=rand();}inline void PushUp(int rt){    size[rt]=size[ch[rt][0]]+size[ch[rt][1]]+counts[rt];}void Rotate(int &rt,int kind){    int y=ch[rt][kind^1];    ch[rt][kind^1]=ch[y][kind];    ch[y][kind]=rt;    PushUp(rt);PushUp(y);    rt=y;}void Insert(int &rt,int v){    if(rt==0)    {        NewNode(rt,v);        return ;    }    if(v==val[rt]) counts[rt]++;    else    {        int kind=(v>val[rt]);        Insert(ch[rt][kind],v);        if(r[ch[rt][kind]]<r[rt])            Rotate(rt,kind^1);    }    PushUp(rt);}int select(int rt,int k){    if(size[ch[rt][0]]>=k) return select(ch[rt][0],k);    if(size[ch[rt][0]]+counts[rt]>=k) return val[rt];    return select(ch[rt][1],k-(size[ch[rt][0]]+counts[rt]));}bool remove(int &rt,int v){    if(rt==0) return false;    if(val[rt]==v)    {        if(counts[rt]>1)            counts[rt]--;        else if(!ch[rt][0]&&!ch[rt][1])        {rt=0;return true;}        else        {            int kind=r[ch[rt][0]]<r[ch[rt][1]];            Rotate(rt,kind);            remove(rt,v);        }        PushUp(rt);        return true;    }    else    {        bool flag=remove(ch[rt][v>val[rt]],v);        PushUp(rt);        return flag;    }    return true;}void Init(){    ch[0][0]=ch[0][1]=0;    size[0]=counts[0]=val[0]=0;    tot=root=0;    r[0]=(1LL<<31)-1;}char str[20];int main(){    //freopen("in.txt","r",stdin);    //freopen("out.txt","w",stdout);    int t,n;    scanf("%d",&t);    while(t--)    {        scanf("%d",&n);        Init();        int cnt=0,v;        while(n--)        {            scanf("%s",str);            scanf("%d",&v);            if(str[0]=='r')            {                if(!remove(root,v))                    printf("Wrong!\n");                else                {                    cnt--;                    if(cnt&1)                        printf("%d\n",select(root,cnt/2+1));                    else if(cnt!=0)                    {                        ll tmp=((ll)select(root,cnt/2)+select(root,cnt/2+1));                        if(tmp&1) printf("%.1lf\n",tmp/2.0);                        else printf("%lld\n",tmp/2);                    }                    else printf("Empty!\n");                }            }            else            {                Insert(root,v);                cnt++;                if(cnt&1)                    printf("%d\n",select(root,cnt/2+1));                else                {                    ll tmp=((ll)select(root,cnt/2)+select(root,cnt/2+1));                    if(tmp&1) printf("%.1lf\n",tmp/2.0);                    else printf("%lld\n",tmp/2);                }                   // printf("%.1lf\n",(select(root,cnt/2)+select(root,cnt/2+1))/2.0);            }        }    }    return 0;}

0 0