【BZOJ 3196】 Tyvj 1730 二逼平衡树

终于填坑!!!!!!
鬼知道以前怎么错的!!
反正现在A了
虽然tyvj还是80 但我相信 那是splay的问题 换做treap就可以(虽然不会....

以后写代码 不要怕 长名变量 取有意义的名字

要以易于维护为第一要求 其次再是美观!!

这次有很多地方可以省略 但为了可维护 就不要怕麻烦 反正比WA掉强!!!

http://pan.baidu.com/s/1nuta3xf 用金币在Tyvj扣的数据 2组大 2组小 我还对着小数据手动排序【捂脸 其实自己通读一遍代码就差不多!

#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <algorithm>
using namespace std;

int init[50000+1];
int seg_num[50000*4];
int num[2000000],ch[2000000][2],fa[2000000],size[2000000],root[50000*4],cnt=0;
int n,m,opt;
int ans;
void vist(int now)
{
	if(now==0) return ;
	cout<<num[now]<<' '<<size[now]<<' '<<num[ch[now][0]]<<' '<<num[ch[now][1]]<<endl;
	vist(ch[now][0]);
	vist(ch[now][1]);
}
void Up(int x)
{
	size[x]=size[ch[x][0]]+size[ch[x][1]]+1;
}
void Rotate(int x,int &rt)
{
	int y=fa[x],z=fa[y];
	if(y==rt) 
	{
		rt=x;
		z=0;///
	}
	else
	{
		ch[z][ ch[z][1]==y ]=x;
	}
	int c=ch[y][1]==x;
	fa[x]=z;fa[y]=x;fa[ch[x][c^1]]=y;
	ch[y][c]=ch[x][c^1];ch[x][c^1]=y;
	Up(y);Up(x);
}
void Splay(int x,int &rt)
{
	while(x!=rt)
	{
		if(fa[x]!=rt) Rotate(fa[x],rt);
		Rotate(x,rt);
	}
}
void Ins(int &now,int k,int last,int &rt)
{
	if(now==0)
	{
		now=++cnt;
		fa[now]=last;
		size[now]=1;
		num[now]=k;
		
		Splay(now,rt);
		return ;
	} 
	if(k>=num[now]) Ins(ch[now][1],k,now,rt);
	else Ins(ch[now][0],k,now,rt);
}
void Build(int now,int L,int R)
{
	for(int i=L;i<=R;i++) Ins(root[now],init[i],root[now],root[now]);
	
	if(L==R) return;
	
	Build(now*2,L,(L+R)/2);
	Build(now*2+1,(L+R)/2+1,R);
}
void Sp_Rank(int now,int k)
{

	if(now==0) return ;
	if(num[now]>=k) 
	{
		Sp_Rank(ch[now][0],k);
	}
	else
	{
		ans+=size[ch[now][0]]+1;
		Sp_Rank(ch[now][1],k);
	}
}
void Seg_Rank(int now,int L,int R,int s,int t,int k)
{
	if(s<=L&&R<=t)
	{
		Sp_Rank(root[now],k);
		return ;
	}
	int mid=(L+R)/2;
	if(s<=mid) Seg_Rank(now*2,L,mid,s,t,k);
	if(mid+1<=t) Seg_Rank(now*2+1,mid+1,R,s,t,k);
}
int Kth(int s,int t,int k)
{
	int L=0,R=100000000;
	int ou;
	while(L<R)
	{
		int mid=(L+R)/2;
		ans=1;
		Seg_Rank(1,1,n,s,t,mid);
		if(ans<=k) ou=mid,L=mid+1;
		else R=mid;
	}
	return ou;
}
int Find(int now,int x)
{
	if(num[now]==x) return now;
	if(x<num[now]) return Find(ch[now][0],x);
	else return Find(ch[now][1],x);
}
void Sp_Change(int &rt,int v,int k)
{
	int loc=Find(rt,v);
	Splay(loc,rt);
	if(ch[rt][0]*ch[rt][1]==0) rt=ch[rt][0]+ch[rt][1];
	else
	{
		int tmp=ch[rt][0];
		while(ch[tmp][1]!=0) tmp=ch[tmp][1];
		Splay(tmp,ch[rt][0]); ch[ch[rt][0]][1]=ch[rt][1];
		size[ch[rt][0]]+=size[ch[rt][1]];
		fa[ch[rt][1]]=ch[rt][0];
		rt=ch[rt][0];
	}
	Ins(rt,k,rt,rt); 
}
void Seg_Change(int now,int L,int R,int pos,int k)
{
	Sp_Change(root[now],init[pos],k);
	if(L==R) return ;
	int mid=(L+R)/2;
	if(pos<=mid) Seg_Change(now*2,L,mid,pos,k);
	else Seg_Change(now*2+1,mid+1,R,pos,k); 
}
void Sp_Before(int now,int k)
{
	if(now==0) return ;
	if(num[now]>=k) Sp_Before(ch[now][0],k);
	else ans=max(ans,num[now]),Sp_Before(ch[now][1],k);
}
void Seg_Before(int now,int L,int R,int s,int t,int k)
{
	if(s<=L&&R<=t) 
	{
		Sp_Before(root[now],k);
		return ;
	}
	int mid=(L+R)/2;
	if(s<=mid) Seg_Before(now*2,L,mid,s,t,k);
	if(mid+1<=t) Seg_Before(now*2+1,mid+1,R,s,t,k);
}
void Sp_After(int now,int k)
{
	if(now==0) return ;
	if(num[now]<=k) Sp_After(ch[now][1],k);
	else ans=min(ans,num[now]),Sp_After(ch[now][0],k);
}
void Seg_After(int now,int L,int R,int s,int t,int k)
{
	if(s<=L&&R<=t) 
	{
		Sp_After(root[now],k);
		return ;
	}
	int mid=(L+R)/2;
	if(s<=mid) Seg_After(now*2,L,mid,s,t,k);
	if(mid+1<=t) Seg_After(now*2+1,mid+1,R,s,t,k);
}
int main()
{
//	freopen("a.in","r",stdin);
	
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d",&init[i]); 
	Build(1,1,n);
	
	int L,R,pos,k;
	while(m--)
	{
		scanf("%d",&opt);
		switch (opt)
		{
			case 1:
				{
					scanf("%d %d %d",&L,&R,&k);
					ans=0;
					Seg_Rank(1,1,n,L,R,k);
					printf("%d\n",ans+1);
					break;
				}
			case 2:
				{
					scanf("%d %d %d",&L,&R,&k);
					printf("%d\n",Kth(L,R,k));
					break;
				}
			case 3:
				{
					scanf("%d %d",&pos,&k);
					Seg_Change(1,1,n,pos,k);
					init[pos]=k;
					break;
				} 
			case 4:
				{
					scanf("%d %d %d",&L,&R,&k);
					ans=-1;
					Seg_Before(1,1,n,L,R,k);
					printf("%d\n",ans);
					break;
				}
			case 5:
				{
					scanf("%d %d %d",&L,&R,&k);
					ans=100000000+1;
					Seg_After(1,1,n,L,R,k);
					printf("%d\n",ans);
					break;
				}
		}
	}
	
	return 0;
}
posted @ 2016-01-14 11:29  sxb_201  阅读(187)  评论(4编辑  收藏  举报