蒟蒻刚学旋转Treap,打个板子练练手。
题目描述
原题来自:NOI 2004
OIER 公司是一家大型专业化软件公司,有着数以万计的员工。作为一名出纳员,我的任务之一便是统计每位员工的工资。这本来是一份不错的工作,但是令人郁闷的是,我们的老板反复无常,经常调整员工的工资。如果他心情好,就可能把每位员工的工资加上一个相同的量。反之,如果心情不好,就可能把他们的工资扣除一个相同的量。我真不知道除了调工资他还做什么其它事情。
工资的频繁调整很让员工反感,尤其是集体扣除工资的时候。一旦某位员工发现自己的工资已经低于了合同规定的工资下界,他就会立刻气愤地离开公司,并且再也不会回来了。每位员工的工资下界都是统一规定的。每当一个人离开公司,我就要从电脑中把他的工资档案删去,同样,每当公司招聘了一位新员工,我就得为他新建一个工资档案。
老板经常到我这边来询问工资情况,他并不问具体某位员工的工资情况,而是问现在工资第 k 多的员工拿多少工资。每当这时,我就不得不对数万个员工进行一次漫长的排序,然后告诉他答案。
好了,现在你已经对我的工作了解不少了。正如你猜的那样,我想请你编一个工资统计程序。怎么样,不是很困难吧?
输入格式
第一行有两个非负整数 n,min, 表示下面有多少条命令,min 表示工资的下界;
接下来的 nnn 行,每行表示一个命令。命令可以是以下四种之一:
名称 | 格式 | 作用 |
---|---|---|
I 命令 | I k |
新建一个工资档案,初始工资为 k。如果某员工的初始工资低于工资下界,他将立刻离开公司。 |
A 命令 | A k |
把每位员工的工资加上 k。 |
S 命令 | S k |
把每位员工的工资扣除 k。 |
F 命令 | F k |
查询第 k 多的工资。 |
I 命令,A 命令,S 命令中,k 是一个非负整数。F 命令中,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
数据范围与提示
对于全部数据,I 命令的条数不超过 10^5,A 命令和 S 命令的总条数不超过 100,F 命令的条数不超过 10^5。
每次工资调整的调整量不超过 10^3,新员工的工资不超过 10^5。
辣鸡题解:
区间修改,可以不必维护,我们只要相对大小。考虑定义line为零工资线(物理重力势能滑稽)。
A x 则所有已入职员工工资相对line增大,即line-=x; 然后注意保持相对大小,此时权值为x+line(因为line是零啊喂)
S x line操作同理,不同在于S会有员工离职,我们需要将这些点删除。
注意题面:S命令不超过100,最坏情况1e8可以试试,暴搜a!
直接从rt进行dfs n,回溯时调用del() logn。 删除复杂度O(nlogn)
注意:
- 有相同工资的,cnt要减成零(因为我直接cnt次logn调用了)
- 读题a,如果I时x低于MIN则不不能算到答案里
- 最后别忘相对到绝对-line
#include<cstdio> #include<cmath> #include<ctime> #include<cstdlib> #include<cstring> #include<algorithm> #define ls tr[x].ch[0] #define rs tr[x].ch[1] #define ds tr[x].ch[d] #define reg register #define inf 0x6fffffff #define F(i,a,b) for(i=a;i<=b;++i) using namespace std; int cnt,rt,line,MIN; struct TREAP{ int ch[2],w,siz,cnt,rd; }tr[1000005]; void up(int x) { tr[x].siz=tr[ls].siz+tr[rs].siz+tr[x].cnt; } void rot(int &x,int d) { int son=ds; ds=tr[son].ch[d^1]; tr[son].ch[d^1]=x; up(x); up(x=son); } void in(int &x,int w) { if(!x) { x=++cnt; // tr[x].siz=tr[x].cnt=1; tr[x].w=w; tr[x].rd=rand(); return; } tr[x].siz++; if(tr[x].w==w){ tr[x].cnt++;return; } int d=w>tr[x].w; in(ds,w); if(tr[x].rd>tr[ds].rd) rot(x,d); } void del(int &x,int w) { if(!x) return; if(tr[x].w==w) { if(tr[x].cnt>1) { tr[x].cnt--; tr[x].siz--; return; } int d=tr[ls].rd>tr[rs].rd; if(ls==0||rs==0) x=ls+rs; else rot(x,d),del(x,w); } else tr[x].siz--,del(tr[x].ch[tr[x].w<w],w); } void dfs(int x) { if(ls) dfs(ls); if(rs) dfs(rs); if(MIN+line>tr[x].w) { int t=tr[x].cnt; // while(t--) del(rt,tr[x].w); } } int kth(int x,int k) { while(x) { if(k<=tr[ls].siz) x=ls; else if(k>tr[x].cnt+tr[ls].siz) k-=tr[x].cnt+tr[ls].siz,x=rs; else return tr[x].w; } } int main() { char s[5]; srand(time(0)); int n; scanf("%d%d",&n,&MIN); reg int x,num=0; while(n--) { scanf("%s %d",s,&x); if(s[0]=='I') { if(x>=MIN) in(rt,x+line),++num; } else if(s[0]=='A') line-=x; else if(s[0]=='S') { line+=x; dfs(rt); } else { if(tr[rt].siz<x) printf("-1 "); else printf("%d ",kth(rt,tr[rt].siz-x+1)-line); } } printf("%d",num-tr[rt].siz); return 0; }