【BZOJ】1503: [NOI2004]郁闷的出纳员（Splay）

http://www.lydsy.com/JudgeOnline/problem.php?id=1503

这题没有看题解就1a了～好开心，，

其实后面去看题解发现他们的都很麻烦，其实有种很简单的做法：

题目要求全体人+和-，那么我们只用开变量m1来累计即可，由于splay的特殊性～我们将下界加上这个变量m1然后插入到splay中作为根，然后把左边的全部去掉即可～～～啦啦啦～。～然后出答案时，减去这个变量m1～

还有我们在插入一个新的工资时，要加上这个变量m1插入，要不然会被直接刷掉。。。。

#include <cstdio>
using namespace std;

class Splay {
public:
	struct node {
		node* ch[2], *fa;
		int key, size;
		node() { ch[0]=ch[1]=fa=0; key=size=0; }
		void pushup() { size=ch[0]->size+ch[1]->size+1; }
		bool d() { return fa->ch[1]==this; }
		void setc(node* c, bool d) { ch[d]=c; c->fa=this; }
	}*null, * root;
	Splay() { 
		null=new node;
		null->ch[0]=null->ch[1]=null->fa=null;
		root=null;
	}
	void rot(node* rt) {
		node* fa=rt->fa; bool d=rt->d();
		fa->fa->setc(rt, fa->d());
		fa->setc(rt->ch[!d], d); 
		rt->setc(fa, !d);
		fa->pushup();
		if(root==fa) root=rt;
	}
	node* newnode(const int &key) {
		node* ret=new node();
		ret->key=key; ret->size=1;
		ret->ch[0]=ret->ch[1]=ret->fa=null;
		return ret;
	}
	void splay(node* rt, node* fa) {
		while(rt->fa!=fa)
			if(rt->fa->fa==fa) rot(rt);
			else rt->d()==rt->fa->d()?(rot(rt->fa), rot(rt)):(rot(rt), rot(rt));
		rt->pushup();
	}
	void insert(const int &key) {
		if(root==null) { root=newnode(key); return; }
		node* t=root;
		while(t->ch[key>t->key]!=null) t=t->ch[key>t->key];
		node* c=newnode(key);
		t->setc(c, key>t->key);
		t->pushup();
		splay(c, null);
	}
	void remove(const int &key) {
		node* t=root;
		while(t!=null && t->key!=key) t=t->ch[key>t->key];
		if(t==null) return;
		splay(t, null);
		node* rt=root->ch[0];
		if(rt==null) rt=root->ch[1];
		else {
			node* m=rt->ch[1];
			while(m!=null && m->ch[1]!=null) m=m->ch[1];
			if(m!=null) splay(m, root);
			rt=root->ch[0];
			rt->setc(root->ch[1], 1);
		}
		delete root;
		root=rt;
		root->fa=null;
		if(root!=null) root->pushup();
	}
	int rank(const int &key, node* rt) {
		if(rt==null) return 0;
		int s=rt->ch[0]->size+1;
		if(key==rt->key) return s;
		if(key>rt->key) return s+rank(key, rt->ch[1]);
		else return rank(key, rt->ch[0]);
	}
	node* select(const int &k, node* rt) {
		if(rt==null) return null;
		int s=rt->ch[0]->size+1;
		if(s==k) return rt;
		if(k>s) return select(k-s, rt->ch[1]);
		else return select(k, rt->ch[0]);
	}
};

int main() {
	int n, m, ans=0, t, m1=0;
	Splay tree;
	scanf("%d%d", &n, &m);
	char str[3];
	while(n--) {
		scanf("%s%d", str, &t);
		if(str[0]=='I') {
			if(t<m) continue;
			tree.insert(t+m1);
		}
		else if(str[0]=='F') {
			if(t>tree.root->size) printf("-1\n");
			else {
				printf("%d\n", tree.select(tree.root->size-t+1, tree.root)->key-m1);
			}
		}
		else if(str[0]=='S') {
			m1+=t;
			tree.insert(m+m1);
			ans+=tree.root->ch[0]->size;
			tree.root=tree.root->ch[1];
			tree.root->fa=tree.null;
		}
		else if(str[0]=='A') {
			m1-=t;
		}
	}
	printf("%d\n", ans);
	return 0;
}

 

 

---

 

 

Description

OIER公司是一家大型专业化软件公司，有着数以万计的员 工。作为一名出纳员，我的任务之一便是统计每位员工的工资。这本来是一份不错的工作，但是令人郁闷的是，我们的老板反复无常，经常调整员工的工资。如果他 心情好，就可能把每位员工的工资加上一个相同的量。反之，如果心情不好，就可能把他们的工资扣除一个相同的量。我真不知道除了调工资他还做什么其它事情。 工资的频繁调整很让员工反感，尤其是集体扣除工资的时候，一旦某位员工发现自己的工资已经低于了合同规定的工资下界，他就会立刻气愤地离开公司，并且再也 不会回来了。每位员工的工资下界都是统一规定的。每当一个人离开公司，我就要从电脑中把他的工资档案删去，同样，每当公司招聘了一位新员工，我就得为他新 建一个工资档案。老板经常到我这边来询问工资情况，他并不问具体某位员工的工资情况，而是问现在工资第k多的员工拿多少工资。每当这时，我就不得不对数万 个员工进行一次漫长的排序，然后告诉他答案。好了，现在你已经对我的工作了解不少了。正如你猜的那样，我想请你编一个工资统计程序。怎么样，不是很困难 吧？

Input

Output

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

Sample Input

9 10
I 60
I 70
S 50
F 2
I 30
S 15
A 5
F 1
F 2

Sample Output

10
20
-1
2

HINT

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

Source

Splay