【TJOI2019】甲苯先生的滚榜【平衡树】

传送门

练手题，两个关键字随便弄一下就行。不过我好久没打这东西以至于出了好些bug。

顺带来记录一下treap怎么写。

1.大根堆最好（我也不知道为啥啊我小根堆就错了）

2.insert：在二叉查找的基础上移动，移动到空新建。每次查儿子是否rnd大于自己，如果是挪上来。

3.delete：二叉查找移动，如果查到分情况讨论。①：有多个，直接减少。②只有一个，将左或者右儿子按照规则挪上来，将自己挪下去，挪到挪不动就删掉。

4.qrank：查询名次。可以左就左，查到返回左size+1（本题特殊），反之返回左size+我cnt+右查找。

5.前驱后继：二叉查找移动，如果找到本值，左右右右或者右左左左，反之不断逼近更新答案。

6.记得先加一个inf和一个-inf维持树的存在性。rotate别忘了打地址符（否则修改不了），记得srand，，然后没了

 

#include<bits/stdc++.h>
using namespace std;
#define in read()
#define inf (0x3f3f3f3f)
unsigned int in{
	unsigned int cnt=0,f=1;char ch=0;
	while(!isdigit(ch)){
		ch=getchar();if(ch=='-')f=-1;
	}
	while(isdigit(ch)){
		cnt=cnt*10+ch-48;
		ch=getchar();
	}return cnt*f;
}
struct node{
	int ch[2],rnd,size,cnt,ac,tim;
	bool operator <(const node &b){
		if(ac==b.ac){
			return tim<b.tim;
		}return ac<b.ac;
	}
}t[1000003];
unsigned n,m,seed,last=7;
int ac[1000003],tim[1000003];
int root;
typedef unsigned int ui;
    ui randNum(ui& seed, ui last, const ui m) {
    seed = seed * 17 + last;
    return seed % m + 1;
}int tot;
int New(int a,int tt){
	++tot;
	t[tot].ch[0]=t[tot].ch[1]=0;
	t[tot].rnd=rand();t[tot].cnt=t[tot].size=1;t[tot].ac=a;t[tot].tim=tt;
	return tot;
}
void pushup(int u){
	t[u].size=t[u].cnt;
	if(t[u].ch[0])t[u].size+=t[t[u].ch[0]].size;
	if(t[u].ch[1])t[u].size+=t[t[u].ch[1]].size;
}
void build(){
	tot=0;root=New(inf,-inf);t[root].ch[0]=New(-inf,inf);pushup(root);
}
void rotate(int &u,int k){//cout<<"rotate "<<u<<" "<<k<<endl;
	int w=t[t[u].ch[k]].ch[k^1];int x=t[u].ch[k];//cout<<"w "<<w<<endl;
	t[u].ch[k]=w;t[x].ch[k^1]=u;u=x;
	pushup(t[u].ch[k^1]);pushup(u);
}
void insert(int &rt,int Ac,int Tim){
	//cout<<"ins "<<rt<<endl;
	if(!rt){
		rt=New(Ac,Tim);return;
	}
	if(Ac==t[rt].ac&&Tim==t[rt].tim)t[rt].cnt++;
	else{
		int d=0;
		if(t[rt].ac<Ac)d=1;
		if(t[rt].ac==Ac&&t[rt].tim>Tim)d=1;
		insert(t[rt].ch[d],Ac,Tim);
		if(t[rt].rnd<t[t[rt].ch[d]].rnd)rotate(rt,d);
	}
	pushup(rt); 
}
void Delete(int &rt,int Ac,int Tim){
	if(!rt)return;
	if(t[rt].ac==Ac&&t[rt].tim==Tim){
		if(t[rt].cnt>1){
			t[rt].cnt--;pushup(rt);return;
		}
		if(t[rt].ch[0]||t[rt].ch[1]){
			if(!t[rt].ch[1]||t[t[rt].ch[0]].rnd>t[t[rt].ch[1]].rnd){
				rotate(rt,0);Delete(t[rt].ch[1],Ac,Tim);
			}else{
				rotate(rt,1);Delete(t[rt].ch[0],Ac,Tim);
			}
			pushup(rt);
		}
		else rt=0;return;
	}
	if(Ac>t[rt].ac)Delete(t[rt].ch[1],Ac,Tim);
	else if(Ac==t[rt].ac){
		if(Tim<t[rt].tim)Delete(t[rt].ch[1],Ac,Tim);
		else Delete(t[rt].ch[0],Ac,Tim);
	}else Delete(t[rt].ch[0],Ac,Tim);
	pushup(rt);
}
int qrank(int rt,int Ac,int Tim){

	if(!rt)return 0;
	if(t[rt].ac==Ac&&t[rt].tim==Tim)return t[t[rt].ch[0]].size+t[rt].cnt;
	if(Ac<t[rt].ac)return qrank(t[rt].ch[0],Ac,Tim);
	if(Ac==t[rt].ac)if(Tim>t[rt].tim)return qrank(t[rt].ch[0],Ac,Tim);
	return t[t[rt].ch[0]].size+t[rt].cnt+qrank(t[rt].ch[1],Ac,Tim);
}
void debug(int u){
	if(t[u].ch[0])debug(t[u].ch[0]);
	cout<<u<<" "<<t[u].ac<<" "<<t[u].tim<<" "<<t[u].cnt<<" "<<t[u].size<<endl;
	if(t[u].ch[1])debug(t[u].ch[1]); 
}
signed main(){
	int T=in;srand(time(0));
	while(T--){
		for(int i=0;i<=1000000;i++)t[i].ch[0]=t[i].ch[1]=0;
		build();memset(ac,0,sizeof(ac));memset(tim,0,sizeof(tim));
		n=in;m=in;seed=in;//debug(root);
		while(m--){
			int x=randNum(seed,last,n);int y=randNum(seed,last,n);
//			cout<<"x,y "<<x<<" "<<y<<endl;
			Delete(root,ac[x],tim[x]);
//			cout<<"######  before   #########"<<endl;
//			debug(root);
			ac[x]++;tim[x]+=y;
			//cout<<"ac,tim "<<" "<<ac[x]<<" "<<tim[x]<<endl;
			insert(root,ac[x],tim[x]);
//			cout<<"########  after   #########"<<endl;
//			debug(root);
			last=t[root].size-1-qrank(root,ac[x],tim[x]);
			cout<<last<<'\n';
		}
	}
	return 0;
}