hdu 4441 Queue Sequence(伸展树)

题意:模拟一个队列序列,将i插入队列操作表示为+i,将i弹出队列表示为-i。有三个操作:

         ①insert p:找出将当前队列中最小的没出现过的正整数,将+i插入到p位置,然后将-i插入到尽量靠右边的合法位置(合法即这个序列对队列的操作是合法的)。

         ②remove i:将+i和-i移出队列。

         ③query i:求+i和-i之间的和。


思路:愉快地写了伸展树~  这题主要是插入操作,首先选一个最小的没出现过的正整数,这个用个优先队列维护就好了,队列为空时添加一个数,删除i的时候把它添加到队列就行了,插入+i也没问题,插入-i的条件是尽量靠后,根据队列的性质,假设x是p前面出现的正数的个数,那么要在第x+1个负数之前插入-i。新建结点的时候记录一下i的结点编号就能快速找到i了。


代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#include<bitset>
#include<set>
#define inf 0x3f3f3f3f
#define Inf 0x3FFFFFFFFFFFFFFFLL
#define eps 1e-6
#define pi acos(-1.0)
using namespace std;
typedef long long ll;
const int maxn=300000+10;
int ch[maxn][2],pre[maxn],val[maxn],size[maxn];
int pos[maxn][2],pcnt[maxn][2],tot,root;
ll sum[maxn];
priority_queue<int,vector<int>,greater<int> >q;
void NewNode(int &rt,int fa,int v)
{
    rt=++tot;
    ch[rt][0]=ch[rt][1]=0;
    pre[rt]=fa;size[rt]=1;
    val[rt]=sum[rt]=v;
    pcnt[rt][v<0]=1;
    pcnt[rt][v>0]=0;
    pos[abs(v)][v<0]=rt;
}
void PushUp(int rt)
{
    size[rt]=size[ch[rt][0]]+size[ch[rt][1]]+1;
    sum[rt]=sum[ch[rt][0]]+sum[ch[rt][1]]+val[rt];
    pcnt[rt][0]=pcnt[ch[rt][0]][0]+pcnt[ch[rt][1]][0]+(val[rt]>0);
    pcnt[rt][1]=pcnt[ch[rt][0]][1]+pcnt[ch[rt][1]][1]+(val[rt]<0);
}
void Rotate(int x,int kind)
{
    int y=pre[x];
    ch[y][kind^1]=ch[x][kind];
    pre[ch[x][kind]]=y;
    if(pre[y]) ch[pre[y]][ch[pre[y]][1]==y]=x;
    pre[x]=pre[y];
    ch[x][kind]=y;
    pre[y]=x;
    PushUp(y);
}
void Splay(int rt,int goal)
{
    while(pre[rt]!=goal)
    {
        int y=pre[rt];
        if(pre[y]==goal)
            Rotate(rt,ch[y][0]==rt);
        else
        {
            int kind=ch[pre[y]][0]==y;
            if(ch[y][kind]==rt)
            {
                Rotate(rt,kind^1);
                Rotate(rt,kind);
            }
            else
            {
                Rotate(y,kind);
                Rotate(rt,kind);
            }
        }
    }
    PushUp(rt);
    if(goal==0) root=rt;
}
void RotateTo(int k,int goal)
{
    int rt=root;
    while(size[ch[rt][0]]!=k)
    {
        if(size[ch[rt][0]]>k)
            rt=ch[rt][0];
        else
        {
            k-=(size[ch[rt][0]]+1);
            rt=ch[rt][1];
        }
    }
    Splay(rt,goal);
}
void Insert(int k,int v)
{
    RotateTo(k,0);
    RotateTo(k+1,root);
    NewNode(ch[ch[root][1]][0],ch[root][1],v);
    PushUp(ch[root][1]);
    PushUp(root);
}
void Del(int rt)
{
    Splay(rt,0);
    if(!ch[rt][0])
    {
        root=ch[rt][1];
        pre[root]=0;
        return ;
    }
    int rp=ch[rt][0];
    while(ch[rp][1]) rp=ch[rp][1];
    Splay(rp,root);
    ch[rp][1]=ch[rt][1];
    pre[ch[rt][1]]=rp;
    pre[rp]=0;
    root=rp;
    PushUp(root);
}
ll Query(int L,int R)
{
    Splay(L,0);
    Splay(R,root);
    return sum[ch[ch[root][1]][0]];
}
int Find(int p)
{
    RotateTo(p+1,0);
    int k=pcnt[ch[root][0]][0];
    int rt=root;
    while(true)
    {
        if(pcnt[ch[rt][0]][1]>=k)
            rt=ch[rt][0];
        else if(pcnt[ch[rt][0]][1]+(val[rt]<0)==k)
            break;
        else
        {
            k-=pcnt[ch[rt][0]][1]+(val[rt]<0);
            rt=ch[rt][1];
        }
    }
    Splay(rt,0);
    return size[ch[root][0]]-1;
}
void Init()
{
    pre[0]=ch[0][0]=ch[0][1]=0;
    size[0]=sum[0]=pcnt[0][0]=pcnt[0][1]=0;
    tot=root=0;
    NewNode(root,0,1);
    NewNode(ch[root][1],root,-1);
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("hack.out","w",stdout);
    int n,x,tcase=0;
    char str[15];
    while(~scanf("%d",&n))
    {
        printf("Case #%d:\n",++tcase);
        Init();
        while(!q.empty()) q.pop();
        int now=1;
        while(n--)
        {
            scanf("%s%d",str,&x);
            if(str[0]=='i')
            {
                if(q.empty()) q.push(now++);
                int v=q.top();
                q.pop();
                Insert(x,v);
                int ps=Find(x);
                Insert(ps,-v);
            }
            else if(str[0]=='r')
            {
                int a=pos[x][0];
                Del(a);
                a=pos[x][1];
                Del(a);
                q.push(x);
            }
            else
            {
                int L=pos[x][0],R=pos[x][1];
                printf("%I64d\n",Query(L,R));
            }
        }
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值