[NOI2004]郁闷的出纳员 (Splay)

最新推荐文章于 2022-09-07 19:12:29 发布

w20810

最新推荐文章于 2022-09-07 19:12:29 发布

阅读量471

点赞数

分类专栏： ACM-数据结构文章标签： Splay

本文链接：https://blog.youkuaiyun.com/w20810/article/details/50478523

版权

ACM-数据结构专栏收录该内容

87 篇文章

订阅专栏

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

题意：

1503: [NOI2004]郁闷的出纳员

Time Limit: 5 Sec Memory Limit: 64 MB
Submit: 8962 Solved: 3108
[ Submit][ Status][ Discuss]

Description

OIER公司是一家大型专业化软件公司，有着数以万计的员工。作为一名出纳员，我的任务之一便是统计每位员工的工资。这本来是一份不错的工作，但是令人郁闷的是，我们的老板反复无常，经常调整员工的工资。如果他心情好，就可能把每位员工的工资加上一个相同的量。反之，如果心情不好，就可能把他们的工资扣除一个相同的量。我真不知道除了调工资他还做什么其它事情。工资的频繁调整很让员工反感，尤其是集体扣除工资的时候，一旦某位员工发现自己的工资已经低于了合同规定的工资下界，他就会立刻气愤地离开公司，并且再也不会回来了。每位员工的工资下界都是统一规定的。每当一个人离开公司，我就要从电脑中把他的工资档案删去，同样，每当公司招聘了一位新员工，我就得为他新建一个工资档案。老板经常到我这边来询问工资情况，他并不问具体某位员工的工资情况，而是问现在工资第k多的员工拿多少工资。每当这时，我就不得不对数万个员工进行一次漫长的排序，然后告诉他答案。好了，现在你已经对我的工作了解不少了。正如你猜的那样，我想请你编一个工资统计程序。怎么样，不是很困难吧？

Input

Output

输出文件的行数为F命令的条数加一。对于每条F命令，你的程序要输出一行，仅包含一个整数，为当前工资第k多的员工所拿的工资数，如果k大于目前员工的数目，则输出-1。输出文件的最后一行包含一个整数，为离开公司的员工的总数。

Sample Input

9 10
I 60
I 70
S 50
F 2
I 30
S 15
A 5
F 1
F 2

Sample Output

10
20
-1
2

HINT

I命令的条数不超过100000 A命令和S命令的总条数不超过100 F命令的条数不超过100000 每次工资调整的调整量不超过1000 新员工的工资不超过100000

Source

Splay

[ Submit][ Status][ Discuss]
HOME Back

分析：Splay~

用一个变量buf保存所有员工需要变动的数值，每次加一个员工的时候只需将工资减去buf再插进树中即可。

删除工资小于min的员工也很简单，先增加一个员工使其工资刚好大于最低工资，由于Splay的特性，新增的员工会在根节点，然后直接把左子树删掉即可。再删掉根节点。为了防止数值冲突，我把所有工资扩大了1倍，方便增加这个临时员工。

用Splay解决这道题，太爽了，啊哈哈。。

吐槽：莫名奇妙RE了一天。。。。

代码：

#include <bits/stdc++.h>
using namespace std;

typedef long long LL;
typedef unsigned long long ULL;
const LL INF = 1E9+9;
const int MI = ~0u>>1;
const int maxn = 2e6+7;

int son[maxn][2],fa[maxn],sz[maxn],key[maxn];
int root,cnt;

inline void Init()
{
	fa[0]=sz[0]=son[0][0]=son[0][1]=cnt=root=0;
}
inline void pushup(int rt)
{
	sz[rt]=sz[son[rt][0]]+sz[son[rt][1]];
	if(rt)	++sz[rt];
}
inline int newnode(int f,int v)
{
	++cnt;
	fa[cnt]=f;
	key[cnt]=v;
	son[cnt][0]=son[cnt][1]=0;
	sz[cnt]=1;
	return cnt;
}
inline void Rotate(int r,int kind)
{
	int y=fa[r];
	son[y][kind^1]=son[r][kind];
	fa[son[r][kind]] = y;
	if(fa[y]!=0)
		son[fa[y]][son[fa[y]][1]==y] = r;
	fa[r]=fa[y];
	son[r][kind]=y;
	fa[y]=r; 
	pushup(y);
}
void Splay(int r,int goal)
{
	while(fa[r]!=goal)
	{
		if(goal==fa[fa[r]])
			Rotate(r,son[fa[r]][0]==r);
		else
		{
			int y=fa[r];
			int kind=son[fa[y]][0]==y;
			if(son[y][kind]==r)
			{
				Rotate(r,kind^1);
				Rotate(r,kind);
			}
			else
			{
				Rotate(y,kind);
				Rotate(r,kind);
			}
		}
	}
	if(0==goal)
		root=r;
	pushup(r);
}
int get_pre(int r)
{
	if(0==son[r][0])
		return -1;
	r=son[r][0];
	while(son[r][1])
		r=son[r][1];
	return r;
}
int get_next(int r)
{
	if(0==son[r][1])
		return -1;
	r=son[r][1];
	while(son[r][1])
		r=son[r][1];
	return r;
}

void Insert(int v)
{
	if(0==root)
	{
		root=newnode(0,v);
		return ;
	}
	int r = root;
	while(son[r][key[r]<v])
		r=son[r][key[r]<v];
	son[r][key[r]<v]=newnode(r,v);
	Splay(son[r][key[r]<v],0);
	pushup(root);
}

void DelteBehind(int v)
{
	Insert(v+1);
	son[root][0]=0;
	root=son[root][1];
	fa[root]=0;
}
int Kth(int k)
{
	int r=root;
	while(r)
	{
		int lnum=sz[son[r][0]];
		if(lnum+1==k)
		{
			Splay(r,0);
			return key[r];
		}
		else if(k<=lnum)
			r=son[r][0];
		else
		{
			r=son[r][1];
			k=k-lnum-1;
		}
	}
	return MI;
}

int Min,buf,ans;
inline void I(int k)
{
	if(k<Min) return ;
	k-=buf;
	Insert(k);
}
inline void A(int k)
{
	buf+=k;
}
inline void S(int k)
{
	buf-=k;
	int f=Min-buf;
	int tot1=sz[root];
	DelteBehind(f-2);
	int tot2=sz[root];
	ans+=tot1-tot2;
}
inline void F(int k)
{
	sz[root]<k?puts("-1"):printf("%d\n",((Kth(sz[root]-k+1)+buf)>>1));
}

int main()
{
	int n;
	while(scanf("%d%d",&n,&Min)!=EOF)
	{
		Init();
		Min=Min<<1;
		ans=buf=0;
		char op[2];
		int k;
		while(n--)
		{
			scanf("%s%d",op,&k);
			switch(op[0])
			{
				case 'I':I(k<<1);break;
				case 'A':A(k<<1);break;
				case 'S':S(k<<1);break;
				case 'F':F(k);break;
			}
		}
		printf("%d\n",ans);
	}
	return 0;
}