P2596 [ZJOI2006]书架

题目

题目

思路

很恶心的FHQTreap，话说出题人好像是打算考Splay来着？没事，反正出题人可能卡不了FHQTreap
连权值线段树，文艺平衡树，普通Treap都能过，这题写FHQ已经很给出题人面子了……
相比普通FHQTreap不能说毫无变化，至少也是一模一样……
唯一一个稍微有点用的思想就是用个map（数组也一样）来记录编号与平衡树里节点编号对应关系。
然后这题需要一个求rank的函数
总体来说就是一堆封装程度极高的玩意拼在一起……
记个知识：函数参数可以在定义时设初始值，效果未知，这里也是随便乱搞一下发现的，这里进行WJ同学喜（怨）闻（声）乐（载）见（道）的实验，发现暂时有一个效果：传递参数可以不传这些设了初始值的，感觉传不传都一样？其余效果未知，就当个古怪知识存在这，有空再研究
直接上代码吧
code:

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<string>
#include<iostream>
#include<cstdlib>
using namespace std;
inline int read()
{
    int ret,c,f=1;
    while (((c=getchar())> '9'||c< '0')&&c!='-');
    if (c=='-') f=-1,ret=0;
    else ret=c-'0';
    while ((c=getchar())>='0'&&c<='9') ret=ret*10+c-'0';
    return ret*f;
}
struct f{
	int val,key,l,r,fa,siz,lazy;//真值，权值，左右儿子，父亲，子树大小，特殊标记这里指翻转 
} tree[100006];
int cnt,n,m,root,x1,x2,x3,x4,x,z;
int id[100006];
int New(int x)//建新点，传真值 
{
	tree[++cnt].val=x;
	tree[cnt].siz=1;
	tree[cnt].key=rand();
	id[x]=cnt;
	return cnt;
}
void down(int id)//标记下传 
{
	if (!tree[id].lazy) return;
	swap(tree[id].l,tree[id].r);
	if (tree[id].l) tree[tree[id].l].lazy^=1;
	if (tree[id].r) tree[tree[id].r].lazy^=1;
	tree[id].lazy=0;
	return;
}
void up(int id)//上传 
{
	tree[id].siz=tree[tree[id].l].siz+tree[tree[id].r].siz+1;
	return;
}
void split(int id,int siz,int &l,int &r,int fat_l=0,int fat_r=0)//分裂，把从1开始长度为siz的部分拆出来 
{
	if (id==0)
	{
		l=r=0;
		return;
	}
	down(id);
	if (siz>tree[tree[id].l].siz)//递归左儿子 
	{
		tree[id].fa=fat_l;
		l=id;
		split(tree[l].r,siz-1-tree[tree[l].l].siz,tree[l].r,r,id,fat_r);
	}
	else//递归右儿子 
	{
		tree[id].fa=fat_r;
		r=id;
		split(tree[r].l,siz,l,tree[r].l,fat_l,id);
	}
	up(id);
	return;
}
int merge(int l,int r)//合并，传入根节点编号 
{
	if (l==0||r==0)
	{
		return l+r;
	}
	if (tree[l].key<tree[r].key)//维护小根堆性质
	{
		down(l);
		tree[l].r=merge(tree[l].r,r);
		tree[tree[l].r].fa=l;
		up(l);
		return l;
	}
	else
	{
		down(r);
		tree[r].l=merge(l,tree[r].l);
		tree[tree[r].l].fa=r;
		up(r);
		return r;
	}
}
int GR(int id)//求排名 
{
	int rak=tree[tree[id].l].siz+1,idf=id;
	while (id!=root&&idf)
	{
		if (tree[tree[idf].fa].r==idf) rak+=tree[tree[tree[idf].fa].l].siz+1;
		idf=tree[idf].fa;
	}
	return rak;
}
int main()
{
	n=read(),m=read();
	for (int i=1;i<=n;i++)
	{
		root=merge(root,New(read()));
	}
	while (m--)
	{
		string opt;
		cin>>opt;
		if (opt=="Top")
		{
			x=read();
			x=GR(id[x]);
			split(root,x,x1,x2);
			split(x1,x-1,x1,x3);
			root=merge(x3,merge(x1,x2));
		}
		if (opt=="Bottom")
		{
			x=read();
			x=GR(id[x]);
			split(root,x,x1,x2,0);
			split(x1,x-1,x1,x3,0);
			root=merge(x1,merge(x2,x3));
		}
		if (opt=="Ask")
		{
			x=read();
			x=GR(id[x])-1;
			printf("%d\n",x);
		}
		if (opt=="Query")
		{
			x=read();
			split(root,x,x1,x2);
			x3=x1;
			while (tree[x3].r) x3=tree[x3].r;
			printf("%d\n",tree[x3].val);
			root=merge(x1,x2);
		}
		if (opt=="Insert")
		{
			x=read(),z=read();
			x=GR(id[x]);
			if (z==0) continue;
			if (z==1)
			{
				split(root,x+1,x1,x2);
				split(x1,x,x3,x1);
				split(x3,x-1,x4,x3);
			}
			else
			{
				split(root,x,x1,x2);
				split(x1,x-1,x3,x1);
				split(x3,x-2,x4,x3);
			}
			root=merge(x4,merge(x1,merge(x3,x2)));
		}
	}
	return 0;
}