[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。

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

输入样例#1: 
9 10
I 60
I 70
S 50
F 2
I 30
S 15
A 5
F 1
F 2
输出样例#1: 
10
20
-1
2

说明

I命令的条数不超过100000

A命令和S命令的总条数不超过100

F命令的条数不超过100000

每次工资调整的调整量不超过1000

新员工的工资不超过100000

这道题是一道经典的平衡树练习题,题中一共有四种操作:插入,查询,修改(上调和下降)。

这里面唯一和一般的treap不一样的操作就是修改,如果每一次都把每个数加上或减掉一个值显然是不行的,但我们发现每次修改都是对全体修改,那么我们就可以不进行修改,相对的把工资下界调整,这样也就相当于修改了每个员工的工资。

但是如果在某次涨工资之后来了一名员工,他并没有赶上涨工资,如果他的工资大于初始工资下界(这样才能进来)但却小于当前工资下界,那么下一次降工资后他本来不会跳槽(下降之后仍大于初始工资下界),却因为小于当前工资下界,那他到底会不会跳槽呢?依据题意是不会的,但如果按上述做法就是会跳槽的。那么就要在他刚进公司时我们要先给他加上之前涨的工资(使他工资一定大于当前工资下界),在查询到他时再减掉就可以了。

工资下降相当于工资下界上调,这样就有可能有人会跳槽,因为I命令不超过10^5,树中最多不会超过10^5个节点,删除点个数也不会超过这些,所以暴力删除就好。

如果查前驱函数是void函数要记得每次查询之后把answer归零。

  1 #include<cstdio>
  2 #include<algorithm>
  3 #include<iostream>
  4 #include<cmath>
  5 #include<ctime>
  6 #include<cstring>
  7 using namespace std;
  8 int r[100010];
  9 int v[100010];
 10 int w[100010];
 11 int ls[100010];
 12 int rs[100010];
 13 int size[100010];
 14 int tot;
 15 int root;
 16 int n,m;
 17 int del;
 18 int mini;
 19 int x;
 20 char a[10];
 21 int answer;
 22 int ans;
 23 void up(int x)
 24 {
 25     size[x]=size[rs[x]]+size[ls[x]]+w[x];
 26 }
 27 void rturn(int &x)
 28 {
 29     int t;
 30     t=ls[x];
 31     ls[x]=rs[t];
 32     rs[t]=x;
 33     size[t]=size[x];
 34     up(x);
 35     x=t;
 36 }
 37 void lturn(int &x)
 38 {
 39     int t;
 40     t=rs[x];
 41     rs[x]=ls[t];
 42     ls[t]=x;
 43     size[t]=size[x];
 44     up(x);
 45     x=t;
 46 }
 47 void insert_sum(int x,int &i)
 48 {
 49     if(!i)
 50     {
 51         i=++tot;
 52         w[i]=size[i]=1;
 53         v[i]=x;
 54         r[i]=rand();
 55         return ;
 56     }
 57     size[i]++;
 58     if(x==v[i])
 59     {
 60         w[i]++;
 61     }
 62     else if(x>v[i])
 63     {
 64         insert_sum(x,rs[i]);
 65         if(r[rs[i]]<r[i])
 66         {
 67             lturn(i);
 68         }
 69     }
 70     else
 71     {
 72         insert_sum(x,ls[i]);
 73         if(r[ls[i]]<r[i])    
 74         {
 75             rturn(i);
 76         }
 77     }
 78       
 79     return ;
 80 }
 81 void delete_sum(int x,int &i)
 82 {
 83     if(i==0)
 84     {
 85         return ;
 86     }
 87     if(v[i]==x)
 88     {
 89         if(w[i]>1)
 90         {
 91             w[i]--;
 92             size[i]--;
 93             return ;
 94         }
 95         if((ls[i]*rs[i])==0)
 96         {
 97             i=ls[i]+rs[i];
 98         }
 99         else if(r[ls[i]]<r[rs[i]])
100         {
101             rturn(i);
102             delete_sum(x,i);
103         }
104         else
105         {
106             lturn(i);
107             delete_sum(x,i);
108         }
109         return ;
110     }
111     size[i]--;
112     if(v[i]<x)
113     {
114         delete_sum(x,rs[i]);
115     }
116     else
117     {
118         delete_sum(x,ls[i]);
119     }
120     return ;
121 }
122 int ask_sum(int x,int i)
123 {
124     if(i==0)
125     {
126         return 0;
127     }
128     if(x>size[ls[i]]+w[i])
129     {
130         return ask_sum(x-size[ls[i]]-w[i],rs[i]);
131     }
132     else if(size[ls[i]]>=x)
133     {
134         return ask_sum(x,ls[i]);
135     }
136     else
137     {
138         return v[i];
139     }
140 }
141 void ask_front(int x,int i)
142 {
143     if(i==0)
144     {
145         return ;
146     }
147     if(v[i]<x)
148     {
149         answer=i;
150         ask_front(x,rs[i]);
151         return ;
152     }
153     else
154     {
155         ask_front(x,ls[i]);
156         return ;
157     }
158     return ;
159 }
160 int main()
161 {
162     srand(12378);
163     scanf("%d%d",&n,&m);
164     mini=m;
165     for(int i=1;i<=n;i++)
166     {
167         scanf("%s",a+1);
168         scanf("%d",&x);
169         if(a[1]=='I')
170         {
171             if(x>=mini)
172             {
173                 insert_sum(x-del,root);
174             }
175         }
176         else if(a[1]=='A')
177         {
178             m-=x;
179             del+=x;
180         }
181         else if(a[1]=='S')
182         {
183             m+=x;
184             del-=x;
185             answer=0;
186             ask_front(m,root);
187             while(answer!=0)
188             {
189                 delete_sum(v[answer],root);
190                 ans++;
191                 answer=0;
192                 ask_front(m,root);
193                  
194             }
195         }
196         else if(a[1]=='F')
197         {
198             if(size[root]-x+1>0)
199             {
200                 printf("%d\n",ask_sum(size[root]-x+1,root)+del);
201             }
202             else
203             {
204                 printf("-1\n");
205             }
206         }
207     }
208     printf("%d",ans);
209 }

猜你喜欢

转载自www.cnblogs.com/Khada-Jhin/p/8987211.html