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。
输出文件的最后一行包含一个整数,为离开公司的员工的总数。
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命令的条数不超过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 }