洛谷 P1486 [NOI2004]郁闷的出纳员【Treap】

题目描述

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命令的条数加一。

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

输出文件的最后一行包含一个整数，为离开公司的员工的总数。

说明

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

题目分析

也是比较裸的一题
说一下几个坑点吧

初始工资低于标准直接离开的人不计入离开的人总数(为这个找了好久 = =)
删除时记得更新总人数要用该结点的cnt，不要直接- -或++

还有不要吐槽蒟蒻神奇的删除方式
(先记录待删除权值再查找删除什么的 = =)
嗯更新工资也是
蒟蒻直接暴力把所有结点都递归更新了一遍

反正思路特直观就对了= =

#include<iostream>
#include<vector>
#include<algorithm>
#include<queue>
#include<cstring>
#include<cstdio>
using namespace std;int read()
{int f=1,x=0;char ss=getchar();while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}return f*x;
}void print(int x)
{if(x<0){putchar('-');x=-x;}if(x>9)print(x/10);putchar(x%10+'0');
}int n,minn;
int num,ans;
queue<int> q;
struct node
{node* ch[2];int v,r,sum,cnt;node(int v) :v(v) {r=rand();sum=cnt=1;ch[0]=ch[1]=NULL;};int cmp(int x){if(x==v)return -1;return x<v?0:1;}void update(){sum=cnt;if(ch[0]!=NULL) sum+=ch[0]->sum;if(ch[1]!=NULL) sum+=ch[1]->sum;}
};
node* rt=NULL;void rotate(node* &p,int d)
{node* k=p->ch[d^1];p->ch[d^1]=k->ch[d];k->ch[d]=p;p->update(); k->update();p=k;
}void ins(node* &p,int x)
{if(p==NULL){p=new node(x);return;}if(x==p->v){p->sum++;p->cnt++;return;}int d=p->cmp(x);ins(p->ch[d],x);if(p->ch[d]->r < p->r)rotate(p,d^1);p->update();
}void up(node* &p,int x)
{if(p==NULL) return ;p->v+=x; if(p->v<minn) {q.push(p->v);int tot=p->cnt;ans+=tot;num-=tot;//总人数的更新用p->cnt}if(p->ch[0]!=NULL)up(p->ch[0],x);if(p->ch[1]!=NULL)up(p->ch[1],x);
}void del(node* &p,int x){if(p==NULL) return;if(x==p->v){if(p->ch[0]==NULL){node *k=p; p=p->ch[1]; delete(k); }else if(p->ch[1]==NULL){node *k=p; p=p->ch[0]; delete(k); }else{int dd=p->ch[0]->r < p->ch[1]->r ?1 :0;rotate(p,dd); del(p->ch[dd],x);}}else if(x < p->v)del(p->ch[0],x);else del(p->ch[1],x);if(p!=NULL)p->update();
}int kth(node* p,int x)
{//找第k大同之前发过的模板，就不注释了int sum=0;if(p->ch[1]!=NULL) sum=p->ch[1]->sum;if(x<=sum) return kth(p->ch[1],x);else if(x<=sum+p->cnt) return p->v;else return kth(p->ch[0],x-sum-p->cnt);
}int main()
{n=read();minn=read();for(int i=1;i<=n;i++){char ss; scanf("%s",&ss);int x=read();if(ss=='I'){if(x<minn)continue;//这里不用计入离开总人数else ins(rt,x),num++; }else if(ss=='A') up(rt,x);else if(ss=='S') up(rt,-x);else if(ss=='F') {if(x>num) print(-1);else print(kth(rt,x));printf("\n");}while(!q.empty()) {del(rt,q.front());q.pop();}//把之前记录的权值都删除}print(ans);return 0;
}