OIER 公司是一家大型专业化软件公司,有着数以万计的员工。作为一名出纳员,我的任务之一便是统计每位员工的工资。这本来是一份不错的工作,但是令人郁闷的是,我们的老板反复无常,经常调整员工的工资。如果他心情好,就可能把每位员工的工资加上一个相同的量。反之,如果心情不好,就可能把他们的工资扣除一个相同的量。我真不知道除了调工资他还做什么其它事情。
工资的频繁调整很让员工反感,尤其是集体扣除工资的时候,一旦某位员工发现自己的工资已经低于了合同规定的工资下界,他就会立刻气愤地离开公司,并且再也不会回来了。每位员工的工资下界都是统一规定的。每当一个人离开公司,我就要从电脑中把他的工资档案删去,同样,每当公司招聘了一位新员工,我就得为他新建一个工资档案。
老板经常到我这边来询问工资情况,他并不问具体某位员工的工资情况,而是问现在工资第k多的员工拿多少工资。每当这时,我就不得不对数万个员工进行一次漫长的排序,然后告诉他答案。
好了,现在你已经对我的工作了解不少了。正如你猜的那样,我想请你编一个工资统计程序。怎么样,不是很困难吧?
如果某个员工的初始工资低于最低工资标准,那么将不计入最后的答案内。
第一行有两个非负整数 \(n\) 和 \(min\)。\(n\) 表示下面有多少条命令,\(min\) 表示工资下界。
接下来的n行,每行表示一条命令。命令可以是以下四种之一:
\(I\) \(k\) 新建一个工资档案,初始工资为 \(k\)。如果某员工的初始工资低于工资下界,他将立刻离开公司。
\(A\) \(k\) 把每位员工的工资加上 \(k\) 。
\(S\) \(k\) 把每位员工的工资扣除 \(k\)。
\(F\) \(k\) 查询第 \(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\)。
这题很显然用 \(Splay\) , 这是我的 \(Splay\) 第一题。
上网学习了\(Splay\),代码调了一天,现在我只想提醒大家:
上网有风险!!!,我看的博客有很多细节都是错的,导致我调代码调了一天!
千万不看:\(this\) \(and\) \(this\)
相信大家都会\(Splay\)
对于操作\(I\),只要利用\(Insert\)。
对于操作\(A\),只要枚举每一个点,给它加\(k\)。
对于操作\(S\),枚举每一个点,给它减\(k\),再看一下树上有哪些小于\(min\)。
先找\(min\),如果找到的点大于或等于min,那么直接把连向左子树的边删掉,否则,先把它的后继旋到根,再把它和根的边删掉
对于操作\(F\),直接找第\(k\)大。
代码:
#include<cstdio>
#include<iostream>
#include<cmath>
using namespace std;
int size[200000],ff[200000],cnt[200000],ch[200000][2],root=0,tot=0;
int n,m,val[200000];
void update(int x)
{
size[x]=size[ch[x][0]]+size[ch[x][1]]+cnt[x];
}
void rotate(int x)
{
int y=ff[x];
int fl=(ch[y][0]==x);
int z=ch[x][fl];
ch[ff[y]][ch[ff[y]][1]==y]=x;
ff[x]=ff[y];
ch[y][fl^1]=z;
ff[z]=y;
ch[x][fl]=y;
ff[y]=x;
update(y),update(x);
}
void splay(int x,int g)
{
while (ff[x]!=g)
{
int y=ff[x];
int z=ff[y];
if (z!=g) (ch[y][0]==x)^(ch[z][0]==y)?rotate(x):rotate(y);
rotate(x);
}
if (g==0) root=x;
}
void insert(int x)
{
int u=root,fa=0;
while (u!=0&&x!=val[u])
{
fa=u;
u=ch[u][x>val[u]];
}
if (u!=0) cnt[u]++,tot++;
else
{
u=++tot;
if (fa==0) root=u;
else ch[fa][x>val[fa]]=u;
size[u]=1;
val[u]=x;
ff[u]=fa;
cnt[u]=1;
}
splay(u,0);
}
void find(int x)
{
int u=root;
if (u==0) return;
while (ch[u][x>val[u]]!=0&&x!=val[u])
u=ch[u][x>val[u]];
splay(u,0);
}
int nxt(int x,int f)
{
find(x);
int u=ch[root][f];
if (u==0) return -1;
while (ch[u][f^1]!=0) u=ch[u][f^1];
return u;
}
void leave()
{
find(m);
int u=root;
if (val[u]>=m) ff[ch[u][0]]=0,ch[u][0]=0,update(u);
else splay(nxt(val[u],1),0),ch[root][0]=0,update(root),ff[u]=0;
}
int findk(int x)
{
int u=root;
if (size[u]<x) return -1;
while (1)
{
if (x<=size[ch[u][1]]) u=ch[u][1];
else if (x<=size[ch[u][1]]+cnt[u]&&x>=size[ch[u][1]]+1) return u;
else x=x-size[ch[u][1]]-cnt[u],u=ch[u][0];
}
}
int main()
{
scanf("%d%d",&n,&m);
insert(1000000000);
for (register int i=1;i<=n;i++)
{
scanf("\n");
char chrr;
int q;
chrr=getchar();
scanf("%d",&q);
if (chrr=='I'&&q>=m) insert(q);
if (chrr=='A') for (register int j=1;j<=tot;j++) val[j]+=q;
if (chrr=='S')
{
for (register int j=1;j<=tot;j++) val[j]-=q;
leave();
}
if (chrr=='F')
{
int totot=findk(q+1);
if (totot==-1||q>size[root]-1) printf("-1\n");
else printf("%d\n",val[totot]);
}
}
printf("%d\n",tot-size[root]);//tot为总数,size[root]为现在剩下的数
}
原文:https://www.cnblogs.com/nibabadeboke/p/12188920.html