P3391 【模板】文艺平衡树(Splay)

解法

splay的基本操作加上一个区间翻转。

代码

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;

int read()
{
	int x=0,p=1; char ch=getchar();
	while((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
	if(ch=='-') p=-1,ch=getchar();
	while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
	return x*p;
}

int ch[N][2],par[N],val[N],cnt[N],siz[N],rev[N],tot,root;
int n,m;

bool chk(int x) { return ch[par[x]][1]==x; }

void pushup(int x) { siz[x]=siz[ch[x][0]]+siz[ch[x][1]]+cnt[x]; }

void pushdown(int x) 
{
	if(rev[x])
	{
		swap(ch[x][1],ch[x][0]);
		rev[ch[x][0]]^=1;
		rev[ch[x][1]]^=1;
		rev[x]=0;
	}
}

void rotate(int x)
{
	int y=par[x],z=par[y],k=chk(x),w=ch[x][k^1];
	ch[y][k]=w,par[w]=y;
	ch[z][chk(y)]=x,par[x]=z;
	ch[x][k^1]=y,par[y]=x;
	pushup(y); pushup(x);
}

void splay(int x,int goal)
{
	int y,z;
	while(par[x]!=goal)
	{
		y=par[x],z=par[y];
		if(z!=goal)
			chk(x)==chk(y) ? rotate(y):rotate(x);
		rotate(x);
	}
	if(!goal) root=x;
}

void insert(int x)
{
	int u=root,p=0;
	while(u && val[u]!=x) p=u,u=ch[u][x>val[u]];
	if(u) cnt[u]++;
	else 
	{
		u=++tot;
		if(p) ch[p][x>val[p]]=u;
		ch[u][0]=ch[u][1]=0;
		par[u]=p; val[u]=x;
		cnt[u]=siz[u]=1;
	}
	splay(u,0);
}

int kth(int k)
{
	int u=root;
	while(1)
	{
		pushdown(u);
		if(ch[u][0] && k<=siz[ch[u][0]]) u=ch[u][0];
		else if(k>siz[ch[u][0]]+cnt[u]) k-=siz[ch[u][0]]+cnt[u],u=ch[u][1];
		else return u;
	}
}

void reverse(int l,int r)
{
	int L=kth(l),R=kth(r+2);
	splay(L,0),splay(R,L);
	rev[ch[R][0]]^=1; 
}

void output(int x)
{
	pushdown(x);
	if(ch[x][0]) output(ch[x][0]);
	if(val[x] && val[x]<=n) printf("%d ",val[x]);
	if(ch[x][1]) output(ch[x][1]);
}

int main()		
{
	//freopen("data.in","r",stdin);
	//freopen("my.out","w",stdout);
	n=read(),m=read();
	for(int i=0;i<=n+1;i++) insert(i);
	while(m--)
	{
		int x=read(),y=read();
		reverse(x,y);
	}
	output(root);
	
    return 0;
}

这题OJ上全RE,本机能过,调了四个小时,最后发现无返回值的函数打了int,这居然能RE,无语。。。

原文地址:https://www.cnblogs.com/lzqlalala/p/10960002.html