P1486 [NOI2004] 郁闷的出纳员

sarapeng

于 2023-05-26 22:27:16 发布

阅读量93

点赞数 1

文章标签： c++ 笔记算法

本文链接：https://blog.csdn.net/sarapeng/article/details/130894703

版权

[NOI2004] 郁闷的出纳员

题目描述

OIER 公司是一家大型专业化软件公司，有着数以万计的员工。作为一名出纳员，我的任务之一便是统计每位员工的工资。这本来是一份不错的工作，但是令人郁闷的是，我们的老板反复无常，经常调整员工的工资。如果他心情好，就可能把每位员工的工资加上一个相同的量。反之，如果心情不好，就可能把当前在公司的所有员工的工资扣除一个相同的量。我真不知道除了调工资他还做什么其它事情。

工资的频繁调整很让员工反感，尤其是集体扣除工资的时候，一旦某位员工发现自己的工资已经低于了合同规定的工资下界，他就会立刻气愤地离开公司，并且再也不会回来了。每位员工的工资下界都是统一规定的。每当一个人离开公司，我就要从电脑中把他的工资档案删去，同样，每当公司招聘了一位新员工，我就得为他新建一个工资档案。

老板经常到我这边来询问工资情况，他并不问具体某位员工的工资情况，而是问现在工资第 $k$ 多的员工拿多少工资。每当这时，我就不得不对数万个员工进行一次漫长的排序，然后告诉他答案。

好了，现在你已经对我的工作了解不少了。正如你猜的那样，我想请你编一个工资统计程序。怎么样，不是很困难吧？

如果某个员工的初始工资低于最低工资标准，那么将不计入最后的答案内。

输入格式

第一行有两个整数 $n$ 和 $\min$ 。 $n$ 表示下面有多少条命令， $\min$ 表示工资下界。

接下来的 $n$ 行，每行一个字符 $x$ 和一个整数 $k$ ，表示一条命令。命令可以是以下四种之一：

I k 新建一个工资档案，初始工资为 $k$ 。如果某员工的初始工资低于工资下界，他将立刻离开公司。
A k 把每位员工的工资加上 $k$ 。
S k 把每位员工的工资扣除 $k$ 。
F k 查询第 $k$ 多的工资。

在初始时，可以认为公司里一个员工也没有。

输出格式

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

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

请注意，初始工资低于工资下界的员工不算做离开公司的员工。

样例 #1

样例输入 #1

样例输出 #1

提示

数据规模与约定

对于全部的测试点，保证：

I 命令的条数不超过 $10^5$ ；
A 和 S 命令的总条数不超过 $100$ ；
F 命令的条数不超过 $10^5$ ；
每次工资调整的调整量不超过 $10^3$ ；
新员工的工资不超过 $10^5$ 。
$\leq n \leq 3 \times 10^5$ ， $\leq \text{min} \leq 10^9$ ，输入的所有数字均在 $32$ 位带符号整形范围内。

复习一下无旋 $T re a p$

FHQ Treap

特点：所有的操作只用到了分裂和合并两种操作。且所有时间复杂度都为 $O(\log n)$

分裂

将树拆为两棵子树 L，R

以键值为对象进行分裂

x是分裂时的基准值

L为根的子树上所有节点的键值都小于或等于x

R为根的子树上所有节点的键值都大于x。

分治的思想进行分裂

步骤：

当前的节点键值<=x
1. 更新L为当前节点
2. 继续在右子树上进行分裂
当前节点的键值 >x
1. 更新R为当前节点
2. 继续在左子树上进行分裂

void Split(int u,int x,int &L,int &R){//权值分裂 
	//u：当前treap树的根节点 
	//x:分裂的基准值 
	//L、R 分裂后的左右子树的根节点
	if(u==0){ 
		L=R=0;//到达叶子结点直接返回 
		return ;
	}
	
	if(t[u].val<=x){
		L=u;//更新L为当前节点
		Split(t[u].rs,x,t[u].rs,R);
	}else{
		R=u;
		Split(t[u].ls,x,L,t[u].ls);
	}
	Update(u);//更新 
}

合并

int Merge(int L,int R){//根据优先级合并、尽可能保持平衡  称为排序树 
	if(L==0||R==0) return L+R;//任意一棵子树为空的处理 将新树的根选为优先级更大的那个 
	if(t[L].pri > t[R].pri){
		t[L].rs=Merge(t[L].rs,R);
		Update(L);
		return L;
	}else{
		t[R].ls=Merge(L,t[R].ls);
		Update(R);
		return R;
	}
}

模板代码

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int cnt=0,root;//cnt-节点存储到数组中的为止  root 树根
struct node{
	int ls,rs;//左右孩子 
	int val,pri;//键值、优先级
	int size;//以当前节点为根的子树节点总数 
}t[N];

int newNode(int x){
	t[++cnt].val=x;
	t[cnt].pri=rand();
	t[cnt].ls=t[cnt].rs=0;
	t[cnt].size=1;
	return cnt;
}

void Update(int u){
	t[u].size=t[t[u].ls].size+t[t[u].rs].size+1;
}

void Split(int u,int x,int &L,int &R){//权值分裂 
	//u：当前treap树的根节点 
	//x:分裂的基准值 
	//L、R 分裂后的左右子树的根节点
	if(u==0){ 
		L=R=0;//到达叶子结点直接返回 
		return ;
	}
	
	if(t[u].val<=x){
		L=u;//更新L为当前节点
		Split(t[u].rs,x,t[u].rs,R);
	}else{
		R=u;
		Split(t[u].ls,x,L,t[u].ls);
	}
	Update(u);//更新 
}

int Merge(int L,int R){//根据优先级合并、尽可能保持平衡  称为排序树 
	if(L==0||R==0) return L+R;//任意一棵子树为空的处理 将新树的根选为优先级更大的那个 
	if(t[L].pri > t[R].pri){
		t[L].rs=Merge(t[L].rs,R);
		Update(L);
		return L;
	}else{
		t[R].ls=Merge(L,t[R].ls);
		Update(R);
		return R;
	}
}

void Insert(int x){//将x插入Treap中
	int L,R;
	Split(root,x,L,R); //分裂成<=x 和 >x 的两棵树 
	int aa=Merge(L,newNode(x));//合并 L和新节点 
	root=Merge(aa,R);//合并左、右 两个树 
}

void Del(int x){
	int L,R,p;
	Split(root,x,L,R);
	Split(L,x-1,L,p);
	p=Merge(t[p].ls,t[p].rs);
	root=Merge(Merge(L,p),R);
} 

int  Rank(int x){
	int L,R;
	Split(root,x-1,L,R);
	int ans=t[L].size+1;//<x的节点个数加一
	root=Merge(L,R);//重新合并回去 
	return ans;
}

int kth(int u,int k){
	if(k==t[t[u].ls].size+1) return u;//当前节点为第k个，直接返回
	else if(k<=t[t[u].ls].size) return kth(t[u].ls,k); //排名小于左子树节点个数,在左子树上找
	else return kth(t[u].rs,k-t[t[u].ls].size-1);//排名大于左子树节点个数，在有字数上找 
} 

int Precursor(int x){
	int L,R;
	Split(root,x-1,L,R);
	int ans=t[kth(L,t[L].size)].val;
	root=Merge(L,R);
	return ans;
} 

int Successor(int x){
	int L,R;
	Split(root,x,L,R);
	int ans=t[kth(R,1)].val;
	root=Merge(L,R);
	return ans;
}
int main()
{
	srand(time(NULL));
	int n,op,x;
	cin>>n;
	while(n--){
		cin>>op>>x;
		switch(op){
			case 1:Insert(x);break;
			case 2:Del(x);break;
			case 3:printf("%d\n",Rank(x));break;
			case 4:printf("%d\n",t[kth(root,x)].val);break;
			case 5:printf("%d\n",Precursor(x));break;
			case 6:printf("%d\n",Successor(x));break; 
		}
	}
 	return 0;
}

思路

我们再明确一下操作和题目要让我们达成的目的

I操作，插入一个数值，并且要求插入的 $k$ 值>=工资下界
A操作，将所有的数值加上 $k$
S操作，将所有的数值减少 $k$
F操作，寻找第 $k$ 大的元素（从大到小）

重点来了，如果某个员工的工资减去 $\leq$ 工资下界，这个员工将会离职。So，这里包含了一个隐藏条件, 那就是删除元素！

还有一个重点，当一个新员工进入公司时，这名员工是不受前面的工资的加减影响的！，这是一个大难点

完整代码

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int cnt=0,root;//cnt-节点存储到数组中的为止  root 树根
int n,m,add,ans; 
struct node{
	int ls,rs;//左右孩子 
	int val,pri;//键值、优先级
	int size;//以当前节点为根的子树节点总数 
}t[N];

int newNode(int x){
	t[++cnt].val=x;
	t[cnt].pri=rand();
	t[cnt].ls=t[cnt].rs=0;
	t[cnt].size=1;
	return cnt;
}

void Update(int u){
	t[u].size=t[t[u].ls].size+t[t[u].rs].size+1;
}

void Split(int u,int x,int &L,int &R){//权值分裂 
	//u：当前treap树的根节点 
	//x:分裂的基准值 
	//L、R 分裂后的左右子树的根节点
	if(u==0){ 
		L=R=0;//到达叶子结点直接返回 
		return ;
	}
	
	if(t[u].val<=x){
		L=u;//更新L为当前节点
		Split(t[u].rs,x,t[u].rs,R);
	}else{
		R=u;
		Split(t[u].ls,x,L,t[u].ls);
	}
	Update(u);//更新 
}

int Merge(int L,int R){//根据优先级合并、尽可能保持平衡  称为排序树 
	if(L==0||R==0) return L+R;//任意一棵子树为空的处理 将新树的根选为优先级更大的那个 
	if(t[L].pri > t[R].pri){
		t[L].rs=Merge(t[L].rs,R);
		Update(L);
		return L;
	}else{
		t[R].ls=Merge(L,t[R].ls);
		Update(R);
		return R;
	}
}

void Insert(int x){//将x插入Treap中
	int L,R;
	Split(root,x,L,R); //分裂成<=x 和 >x 的两棵树 
	int aa=Merge(L,newNode(x));//合并 L和新节点 
	root=Merge(aa,R);//合并左、右 两个树 
}

void Del(int x){
	//1.分裂成 L,R 
	//L:<=x  R:>x
	int L,R;
	Split(root,x,L,R);
	root=R;
	ans+=t[L].size;
}

int Kth(int k){//从大到小的第k个 
	int u=root;
	while(t[t[u].rs].size!=k-1){
		if(t[t[u].rs].size >= k) u=t[u].rs;
		else{
			k=k-t[t[u].rs].size-1;
			u=t[u].ls;
		}
	}
	return u;
}

int main()
{
	char c;
	int k;
	srand(time(NULL));
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>c>>k;
		if(c=='I'){
			if(k>=m) Insert(k-add);
		}else if(c=='A') add+=k;
		else if(c=='S'){
			add-=k;
			Del(m-1-add);
		}else{
			if(t[root].size<k) printf("-1\n");
			else printf("%d\n",t[Kth(k)].val+add);
		}
	}
	printf("%d\n",ans);
 	return 0;
}

谢谢大家

sarapeng

关注

1
点赞
踩
0

收藏

觉得还不错? 一键收藏
1
评论
P1486 [NOI2004] 郁闷的出纳员

OIER 公司是一家大型专业化软件公司，有着数以万计的员工。作为一名出纳员，我的任务之一便是统计每位员工的工资。这本来是一份不错的工作，但是令人郁闷的是，我们的老板反复无常，经常调整员工的工资。如果他心情好，就可能把每位员工的工资加上一个相同的量。反之，如果心情不好，就可能把当前在公司的所有员工的工资扣除一个相同的量。我真不知道除了调工资他还做什么其它事情。
复制链接

扫一扫