【bzoj1503】[NOI2004]郁闷的出纳员

题目描述:

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

输入:


输出:

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

样例输入:

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

样例输出:

10
20
-1
2

题解:

插入,删除,查第k大那就平衡树什么的乱搞。好像听说线段树什么的也行,反正我是为了练treap才写的。。。

代码:

2016.04.14 treap

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>

#ifdef WIN32
	#define LL "%I64d"
#else
	#define LL "%lld"
#endif

#ifdef CT
	#define debug(...) printf(__VA_ARGS__)
	#define setfile() 
#else
	#define debug(...)
	#define filename ""
	#define setfile() freopen(filename".in", "r", stdin); freopen(filename".out", "w", stdout);
#endif

#define R register
#define getc() (S == T && (T = (S = B) + fread(B, 1, 1 << 15, stdin), S == T) ? EOF : *S++)
#define dmax(_a, _b) ((_a) > (_b) ? (_a) : (_b))
#define dmin(_a, _b) ((_a) < (_b) ? (_a) : (_b))
#define cmax(_a, _b) (_a < (_b) ? _a = (_b) : 0)
#define cmin(_a, _b) (_a > (_b) ? _a = (_b) : 0)
char B[1 << 15], *S = B, *T = B;
inline int FastIn()
{
	R char ch; R int cnt = 0; R bool minus = 0;
	while (ch = getc(), (ch < '0' || ch > '9') && ch != '-') ;
	ch == '-' ? minus = 1 : cnt = ch - '0';
	while (ch = getc(), ch >= '0' && ch <= '9') cnt = cnt * 10 + ch - '0';
	return minus ? -cnt : cnt;
}
const int Ta = 1 << 16 | 3, Tb = 33333331;
int Tc, m, minn;
inline int randint(){return Tc = Tc * Ta + Tb;}
struct Treap
{
	int data, key, size;
	Treap *ls, *rs;
	Treap(int _val):data(_val), key(randint()), size(1), ls(NULL), rs(NULL){}
	inline void update()
	{
		size = 1 + (ls ? ls -> size : 0) + (rs ? rs -> size : 0);
	}
}*root;
inline int Size(Treap *x)
{
	return x ? x -> size : 0;
}
struct Pair
{
	Treap *fir, *sec;
};
Pair Split(Treap *x, int k)
{
	if (!x) return (Pair){NULL, NULL};
	Pair y;
	if (Size(x -> ls) >= k)
	{
		y = Split(x -> ls, k);
		x -> ls = y.sec;
		x -> update();
		y.sec = x;
	}
	else
	{
		y = Split(x -> rs, k - Size(x -> ls) - 1);
		x -> rs = y.fir;
		x -> update();
		y.fir = x;
	}
	return y;
}
Treap *Merge(Treap *a, Treap *b)
{
	if (!a) return b;
	if (!b) return a;
	if (a -> key < b -> key)
	{
		a -> rs = Merge(a -> rs, b);
		a -> update();
		return a;
	}
	else
	{
		b -> ls = Merge(a, b -> ls);
		b -> update();
		return b;
	}
}
int Get(Treap *x, int val)
{
	if (!x) return 0;
	return val < x -> data ? Get(x -> ls, val) : Get(x -> rs, val) + Size(x -> ls) + 1;
}
inline int Find(R int k)
{
	Pair x = Split(root, k - 1);
	Pair y = Split(x.sec, 1);
	Treap *ans = y.fir;
	root = Merge(Merge(x.fir, ans), y.sec);
	return ans -> data;
}
inline void Insert(R int val)
{
	R int k = Get(root, val);
	Pair x = Split(root, k);
	Treap *pre = new Treap(val);
	root = Merge(Merge(x.fir, pre), x.sec);
}
int tag, cnt, leave;
void print(Treap *x)
{
	if (!x) return ;
	print(x -> ls);
	printf("%d ",x -> data );
	print(x -> rs);
}
inline int del(Treap* &x, R int val)
{
	if (!x) return 0;
	if (x -> data >= val)
	{
		R int t = del(x -> ls, val);
		x -> size -= t;
		return t;
	}
	else
	{
		R int t = Size(x -> ls) + 1;
		x = x -> rs;
		return t + del(x, val);
	}
}
int main()
{
//	setfile();
	root = NULL;
	m = FastIn(), minn = FastIn();
	for (R int i = 1; i <= m; ++i)
	{
		R char opt;
		while (opt = getc(), opt < 'A' || opt > 'Z');
		R int k = FastIn();
//		print(root);
//		printf("cnt = %d tag = %d\n",cnt, tag );
		if (opt == 'I')
		{
			if (k >= minn)
				Insert(k - tag), ++cnt;
		}
		else if (opt == 'A') tag += k;
		else if (opt == 'S')
		{
			tag -= k;
			R int tmp = del(root, minn - tag);
			cnt -= tmp;
			leave += tmp;
		}
		else
		{
//			printf("size %d k = %d\n",Size(root), k);
			if (k > Size(root)) puts("-1");
			else printf("%d\n",Find(cnt - k + 1) + tag);
		}
	}
	printf("%d\n",leave );
	return 0;
}



 

posted @ 2016-04-14 13:06  cot  阅读(180)  评论(0编辑  收藏  举报