[NOI 2004]郁闷的出纳员(bzoj 1503)


本文最后更新于:2017年9月8日 早上

题目描述

OIER公司是一家大型专业化软件公司,有着数以万计的员工。作为一名出纳员,我的任务之一便是统计每位员工的工资。这本来是一份不错的工作,但是令人郁闷的是,我们的老板反复无常,经常调整员工的工资。如果他心情好,就可能把每位员工的工资加上一个相同的量。反之,如果心情不好,就可能把他们的工资扣除一个相同的量。我真不知道除了调工资他还做什么其它事情。

工资的频繁调整很让员工反感,尤其是集体扣除工资的时候,一旦某位员工发现自己的工资已经低于了合同规定的工资下界,他就会立刻气愤地离开公司,并且再也不会回来了。每位员工的工资下界都是统一规定的。每当一个人离开公司,我就要从电脑中把他的工资档案删去,同样,每当公司招聘了一位新员工,我就得为他新建一个工资档案。

老板经常到我这边来询问工资情况,他并不问具体某位员工的工资情况,而是问现在工资第k多的员工拿多少工资。每当这时,我就不得不对数万个员工进行一次漫长的排序,然后告诉他答案。

好了,现在你已经对我的工作了解不少了。正如你猜的那样,我想请你编一个工资统计程序。怎么样,不是很困难吧?

输入格式

第一行有两个非负整数n和min。n表示下面有多少条命令,min表示工资下界。

接下来的n行,每行表示一条命令。命令可以是以下四种之一:

名称 格式 作用
I命令 I k 新建一个工资档案,初始工资为k。如果某员工的初始工资低于工资下界,他将立刻离开公司。
A命令 A k 把每位员工的工资加上k
S命令 S k 把每位员工的工资扣除k
F命令 F k 查询第k多的工资

I命令、A命令、S命令中的k是一个非负整数,F命令中的k是一个正整数。

在初始时,可以认为公司里一个员工也没有。

输出格式

输出的行数为F命令的条数加1。

对于每条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

数据范围与提示

id 内容
1 I命令的条数不超过100000
2 A命令和S命令的总条数不超过100
3 F命令的条数不超过100000
4 每次工资调整的调整量不超过1000
5 新员工的工资不超过100000

思路

由于对所有员工都要增减工资,若对每个员工都修改,复杂度高达$O(n)$,这时候,我们可以记录一个基准值now,每次对它修改,用一个Treap维护相对工资,加时直接加,减时del掉工资小于min-now的员工。至于查询第k大,也就是查询排名为t[root].sz-k+1的树。

注意

1.insert时应插入x-now,因为之前工资的增减对才插入的值没影响。

Code:

#include<iostream>
#include<cstdio>
#include<cstdlib>
using namespace std;
const int maxn=1e5+10;
struct node
{
	int l,r,v,w,rd,sz;
	node(){}
	node(int l,int r,int v,int w,int rd,int sz):l(l),r(r),v(v),w(w),rd(rd),sz(sz){}
}t[maxn];
int root=0,sz=0,n,mi,now=0,tot=0,lv=0;
void PushUP(int);
void lturn(int&);
void rturn(int&);
int del(int&,int);
void insert(int&,int);
int getsum(int&,int);
int main()
{
	int x;
	char op[2];
	scanf("%d%d",&n,&mi);
	while(n--)
	{
		scanf("%s%d",op,&x);
		if(op[0]=='I')
		{
			if(x>=mi) insert(root,x-now);
		}
		else if(op[0]=='A') now+=x;
		else if(op[0]=='S')
		{
			now-=x;
			lv+=del(root,mi-now);
		}
		else
		{
			if(x<=t[root].sz) printf("%d\n",getsum(root,t[root].sz-x+1)+now);
			else printf("-1\n");
		}
	}
	printf("%d\n",lv);
	return 0;
}
void PushUP(int k)
{
	t[k].sz=t[t[k].l].sz+t[t[k].r].sz+t[k].w;
}
void lturn(int &k)
{
	int fa=t[k].r;t[k].r=t[fa].l;t[fa].l=k;t[fa].sz=t[k].sz;PushUP(k);k=fa;
}
void rturn(int &k)
{
	int fa=t[k].l;t[k].l=t[fa].r;t[fa].r=k;t[fa].sz=t[k].sz;PushUP(k);k=fa;
}
void insert(int &k,int x)
{
	if(!k)
	{
		k=++sz;
		t[k]=node(0,0,x,1,rand(),1);
		return;
	}
	t[k].sz++;
	if(t[k].v==x) t[k].w++;
	else if(t[k].v>x)
	{
		insert(t[k].l,x);
		if(t[t[k].l].rd<t[k].rd) rturn(k);
	}
	else
	{
		insert(t[k].r,x);
		if(t[t[k].r].rd<t[k].rd) lturn(k);
	}
}
int getsum(int &k,int x)
{
	if(k==0) return 0;
	if(x<=t[t[k].l].sz) return getsum(t[k].l,x);
	else if(x>t[t[k].l].sz+t[k].w) return getsum(t[k].r,x-(t[t[k].l].sz+t[k].w));
	else return t[k].v;
}
int del(int &k,int x)
{
	int tp;
	if(k==0) return 0;
	if(t[k].v<x) return t[t[k].l].sz+t[k].w+del(k=t[k].r,x);
	else return tp=del(t[k].l,x),t[k].sz-=tp,tp;
}

文章作者: SpaceSkyNet
版权声明: 本博客所有文章除特別声明外,均采用 CC BY 4.0 许可协议。转载请注明来源 SpaceSkyNet !
  目录
评论