[省选前衡八题目整理][BZOJ 1503]郁闷的出纳员(Splay)

题目链接

http://www.lydsy.com/JudgeOnline/problem.php?id=1503

思路

嘴巴一下还是很简单的，就是个很裸的数据结构题，要求能够支持删除、加入结点以及查询第k大数。

当然Treap和Splay都能做的啦。。。。记得我第一次做是用Treap过的

但是还是有很多细节要注意的，刚开始写完过了样例就交了，没想到居然WA，后来随手在查询完第k大数时把第k大数splay到根节点上去，发现这样做就过不了样例，调了一下午发现是插入结点o时，由于是&o,而插入完结点o后要做splay操作，o的值会发生变化，而最后要给o做pushup操作，因此要先用一个变量tmpo记录下o原来的值，pushup原来的结点o。。。。

代码

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <algorithm>

#define MAXN 210000

using namespace std;

int val[MAXN],ch[MAXN][2],fa[MAXN],size[MAXN];
int nCount=0,root=0;
int limit; //工资下限
int sum=0; //sum=离开公司的员工总数

void pushup(int o)
{
size[o]=size[ch[o][0]]+size[ch[o][1]]+1;
}

void rot(int x,int &k)
{
int y=fa[x],z=fa[y];
int p,q;
if(ch[y][0]==x) p=0; //!!!!!
else p=1;
q=p^1;
if(y!=k)
{
if(ch[z][0]==y) ch[z][0]=x;
else ch[z][1]=x;
}
else k=x;
fa[y]=x,fa[x]=z;
ch[y][p]=ch[x][q];
fa[ch[x][q]]=y;
ch[x][q]=y;
pushup(y);
pushup(x);
}

void splay(int x,int &k)
{
while(x!=k)
{
int y=fa[x],z=fa[y];
if(y!=k)
{
if((ch[y][0]==x)==(ch[z][0]==y)) rot(y,k);
else rot(x,k);
}
rot(x,k);
}
}

void ins(int &o,int father,int v)
{
int tmpo=o; //!!!!!o是会改变的!!!!要用个tmpo记住原来的o是哪个点
if(!o)
{
o=++nCount;
ch[o][0]=ch[o][1]=0;
fa[o]=father;
val[o]=v;
size[o]=1;
splay(o,root);
return;
}
if(v<val[o]) ins(ch[o][0],o,v);
else ins(ch[o][1],o,v);
pushup(tmpo);
}

void del(int &o,int father)
{
if(!o) return;
if(val[o]>=limit)
del(ch[o][0],o);
else
{
sum+=size[ch[o][0]]+1; //!!!!!
o=ch[o][1]; //o和其左子树全部被删了
fa[o]=father;
//if(!father) root=o;
del(o,father);
}
if(o) pushup(o);
}

int findKth(int o,int k)
{
if(!o) return -1;
if(size[ch[o][0]]+1==k)
{
splay(o,root);
return o;
}
if(size[ch[o][0]]>=k) return findKth(ch[o][0],k); //!!!!!!
return findKth(ch[o][1],k-size[ch[o][0]]-1);
}

int main()
{
int q,m,w=0; //w=所有员工加上的工资大小
scanf("%d%d",&q,&m);
char cmd[10];
while(q--)
{
int k;
scanf("%s%d",cmd,&k);
if(cmd[0]=='I')
{
if(k<m) continue; //!!!!!!
ins(root,0,k-w);
}
else if(cmd[0]=='A') w+=k;
else if(cmd[0]=='S')
{
w-=k;
limit=m-w;
del(root,0);
}
else
{
int tmp=size[root];
if(tmp<k)
{
printf("-1\n");
continue;
}
printf("%d\n",val[findKth(root,tmp-k+1)]+w);
}
}
printf("%d\n",sum);
return 0;
}