BZOJ3506 [Cqoi2014]排序机械臂(离散化+Splay)

题目大意:
给定一个n元素数列,n次操作,第i次要求输出第i小的数的当前位置t,再将区间[i,t]翻转 


【题解】

模板题,记录伸展树中 以每个结点为根的子树的最小值,通过它在logN时间内找到最小值的位置,
然后提取出序列并打上翻转标记,
每次操作结束,将最左边的元素分裂出去,这样每次操作都是找最小值 

本题有重复元素,有些麻烦,但题目只让给出位置,可以先离散化去重再处理 


【代码】

#include<stdio.h>
#include<stdlib.h>
#define INF 1000000
int a[100005],b[100005];
struct Node
{
	Node* ch[2];
	int v,s,f,Min;
	int cmp(int x) const
	{
		if(x==ch[0]->s+1) return -1;
		if(x<=ch[0]->s) return 0;
		return 1;
	}
};
Node *root,*null,*left,*right;
int min(int a,int b)
{
	if(a<b) return a;
	return b;
}
void jh(int* a,int* b)
{
	int t=*a;
	*a=*b;
	*b=t;
}
void init()
{
	null=new Node();
	null->ch[0]=null->ch[1]=NULL;
	null->v=null->s=null->f=0;
	null->Min=INF;
	root=left=right=null;
}
void wh(Node* &o)
{
	o->Min = min( o->v , min(o->ch[0]->Min,o->ch[1]->Min) );
	o->s = o->ch[0]->s + 1 + o->ch[1]->s;
}
void pushdown(Node* &o)
{
	if(o->f==0) return;
	Node* t=o->ch[0];
	o->ch[0]=o->ch[1];
	o->ch[1]=t;
	o->f=0;
	o->ch[0]->f^=1;
	o->ch[1]->f^=1;
}
void build(Node* &o,int left,int right)
{
	int mid=(left+right)/2;
	o=new Node();
	o->ch[0]=o->ch[1]=null;
	o->v=a[mid];
	o->f=0;
	if(left<mid) build(o->ch[0],left,mid-1);
	if(right>mid) build(o->ch[1],mid+1,right);
	wh(o);
}
void xz(Node* &o,int d)
{
	Node* k=o->ch[d^1];
	o->ch[d^1]=k->ch[d];
	k->ch[d]=o;
	wh(o);
	wh(k);
	o=k;
}
void splay(Node* &o,int k)
{
	pushdown(o);
	int d=o->cmp(k),d2,k2;
	if(d==1) k -= o->ch[0]->s + 1;
	if(d!=-1)
	{
		Node* p=o->ch[d];
		pushdown(p);
		k2=k;
		d2=p->cmp(k2);
		if(d2==1) k2 -= p->ch[0]->s + 1;
		if(d2!=-1)
		{
			splay(p->ch[d2],k2);
			if(d==d2) xz(o,d^1);
			else xz(o->ch[d],d2^1);
		}
		xz(o,d^1);
	}
}
void fl(Node* &o,int k,Node* &left,Node* &right)
{
	splay(o,k);
	right=o->ch[1];
	o->ch[1]=null;
	left=o;
	wh(left);
}
Node* hb(Node* &left,Node* &right)
{
	splay(left,left->s);
	left->ch[1]=right;
	wh(left);
	return left;
}
int cx(Node* &o)
{
	pushdown(o);
	if( o->v < min(o->ch[0]->Min,o->ch[1]->Min) ) return o->ch[0]->s+1;
	if( o->ch[0]->Min < o->ch[1]->Min ) return cx(o->ch[0]);
	return o->ch[0]->s + 1 + cx(o->ch[1]);
}
void kp(int low,int high)
{
	int i=low,j=high,mida=a[(i+j)/2],midb=b[(i+j)/2];
	while(i<j)
	{
		while(a[i]<mida||(a[i]==mida&&b[i]<midb)) i++;
		while(a[j]>mida||(a[j]==mida&&b[j]>midb)) j--;
		if(i<=j)
		{
			jh(&a[i],&a[j]);
			jh(&b[i],&b[j]);
			i++;
			j--;
		}
	}
	if(j>low) kp(low,j);
	if(i<high) kp(i,high);
}
int main()
{
	int n,i,t;
	init();
	scanf("%d",&n);
	for(i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		b[i]=i;
	}
	kp(1,n);
	for(i=1;i<=n;i++)
		a[b[i]]=i;
	build(root,1,n);
	for(i=1;i<=n;i++)
	{
		t=cx(root);
		printf("%d ",t+i-1);
		fl(root,t,left,right);
		left->f^=1;
		root=hb(left,right);
		fl(root,1,left,right);
		root=right;
	}
	return 0;
}


题目大意:
给定一个n元素数列,n次操作,第i次要求输出第i小的数的当前位置t,再将区间[i,t]翻转 
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值