码迷,mamicode.com
首页 > 其他好文 > 详细

HDU 4441 Queue Sequence(splay)

时间:2014-10-02 00:18:31      阅读:259      评论:0      收藏:0      [点我收藏+]

标签:blog   http   ar   for   sp   div   c   log   amp   

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4441

题意:一个数列,三种操作:(1)插入:找到没在当前数列中的最小的正整数i,将其插在位置p之后,并将-i插入某个位置使得满足先进先出(i表示进,-i表示出),这个位置尽量靠右;(2)删除:删掉数字i以及-i;(3)询问:查询i和-i之间的数字的和。

思路:对于没在数列中的数字可以用一个set直接维护。i的插入是正常的splay操作。对于-i的插入,我们首先找到i之前有几个正数,比如有x个,那么-i必然是插在第x+1个负数的前面,因此在splay节点中要保存负数的个数。删除和查询都是splay正常操作。

struct node
{
	i64 sum;
	int val;
    int size[2];
	int Size;
    node *c[2],*p;

};


node a[N],*root,*nullNode;
int cnt;


void pushUp(node *p)
{
    if(p==nullNode) return;

	p->Size=1;
	p->sum=p->val;
	p->size[0]=p->size[1]=0;
	if(p->val>0) p->size[1]++;
	else if(p->val<0) p->size[0]++;

	if(p->c[0]!=nullNode)
	{
		p->Size+=p->c[0]->Size;
		p->sum+=p->c[0]->sum;
		p->size[0]+=p->c[0]->size[0];
		p->size[1]+=p->c[0]->size[1];
	}

	if(p->c[1]!=nullNode)
	{
		p->Size+=p->c[1]->Size;
		p->sum+=p->c[1]->sum;
		p->size[0]+=p->c[1]->size[0];
		p->size[1]+=p->c[1]->size[1];
	}
}


int ok(node *p)
{
    return p!=nullNode&&(p!=&a[1])&&(p!=&a[2]);
}


node *newNode(int val,node *p)
{
    node *e=&a[val>0?val:-val+100000];
    e->c[0]=e->c[1]=nullNode;
    e->p=p;
    e->Size=1;
	e->val=e->sum=val;
	e->size[0]=e->size[1]=0;

	if(val>0) e->size[1]++;
	else e->size[0]++;
    return e;
}


void init()
{
    nullNode=new node();
    nullNode->size[0]=nullNode->size[1]=0;
	nullNode->Size=0;
    nullNode->sum=0;
	nullNode->c[0]=nullNode->c[1]=nullNode->p=nullNode;


    root=new node();
	root->c[0]=root->c[1]=root->p=nullNode;
    root->sum=0;
	root->val=0;
	root->size[0]=root->size[1]=0;
	root->Size=1;

	root->c[1]=new node();
	root->c[1]->c[0]=root->c[1]->c[1]=nullNode;
	root->c[1]->p=root;
    root->c[1]->sum=root->c[1]->val=-1;
	root->c[1]->Size=1;
	root->c[1]->size[0]=root->c[1]->size[1]=0;
	root->c[1]->size[0]++;

    pushUp(root);

}




void zig(node *x)
{
    node *p=x->p,*q=p->p;
    p->c[0]=x->c[1];
    if(x->c[1]!=nullNode) x->c[1]->p=p;
    x->c[1]=p;
    p->p=x;
    x->p=q;
    if(q!=nullNode)
    {
        if(q->c[0]==p) q->c[0]=x;
        else q->c[1]=x;
    }
    pushUp(p);
    pushUp(x);
    if(root==p) root=x;
}


void zag(node *x)
{
    node *p=x->p,*q=p->p;
    p->c[1]=x->c[0];
    if(x->c[0]!=nullNode) x->c[0]->p=p;
    x->c[0]=p;
    p->p=x;
    x->p=q;
    if(q!=nullNode)
    {
        if(q->c[0]==p) q->c[0]=x;
        else q->c[1]=x;
    }
    pushUp(p);
    pushUp(x);
    if(root==p) root=x;
}


void splay(node *x,node *goal)
{
    while(x->p!=goal)
    {
        if(x->p->p!=goal)
        {
            if(x->p->p->c[0]==x->p)
            {
                if(x->p->c[0]==x) zig(x->p),zig(x);
                else zag(x),zig(x);
            }
            else
            {
                if(x->p->c[1]==x) zag(x->p),zag(x);
                else zig(x),zag(x);
            }
        }
        else
        {
            if(x->p->c[0]==x) zig(x);
            else zag(x);
        }
    }
}



void select(int k,node *goal)
{
    node *p=root;
    while(k!=p->c[0]->Size+1)
    {
        if(k<=p->c[0]->Size) p=p->c[0];
        else
        {
            k-=1+p->c[0]->Size;
            p=p->c[1];
        }
    }
	splay(p,goal);
}

int curNodeNum;

void insert(int p,int val)
{
    select(p,nullNode);
    select(p+1,root);
    node *e=newNode(val,root->c[1]);
    root->c[1]->c[0]=e;
    splay(e,nullNode);

	curNodeNum++;
}

node *getPre(node *u)
{
	splay(u,nullNode);
	u=u->c[0];
	while(u->c[1]!=nullNode) u=u->c[1];
	return u;
}
node *getNext(node *u)
{
	splay(u,nullNode);
	u=u->c[1];
	while(u->c[0]!=nullNode) u=u->c[0];
	return u;
}




node *getKth0(int k)
{
    node *p=root;
    while(1)
    {
        if(p->c[0]->size[0]>=k) p=p->c[0];
        else if(p->c[0]->size[0]+1==k)
        {
            if(p->val<0) return p;
            k-=p->c[0]->size[0];
            p=p->c[1];
        }
        else
        {
            k-=p->c[0]->size[0];
            if(p->val<0) k--;
            p=p->c[1];
        }
    }
}

void del(node *p)
{
	node *u=getPre(p);
	node *v=getNext(p);
	splay(u,nullNode);
	splay(v,root);
	root->c[1]->c[0]=nullNode;
	splay(root->c[1],nullNode);

	curNodeNum--;
}

set<int> S;
int n;



void deal()
{
	S.clear();
	init();

	curNodeNum=2;

	int i;
	for(i=1;i<=100000;i++) S.insert(i);
	while(n--)
	{
		char op[10];
		int p;
		scanf("%s%d",op,&p);
		if(op[0]==‘i‘)
		{
			int t=*S.begin();
			S.erase(t);
			insert(p+1,t);

			splay(a+t,nullNode);
			int num=0;
			if(root->c[0]!=nullNode) num+=root->c[0]->size[1];
			if(root->val>0) num++;

			node *u=getKth0(num);

			node *v=getPre(u);
			splay(v,nullNode);
			splay(u,root);
			node *tmp=newNode(-t,root->c[1]);
			root->c[1]->c[0]=tmp;
			splay(tmp,nullNode);

			curNodeNum++;

		}
		else if(op[0]==‘r‘)
		{
			S.insert(p);
			del(a+p);
			del(a+p+100000);
		}
		else
		{
			node *u=a+p;
			node *v=a+p+100000;
			splay(u,nullNode);
			splay(v,root);
			i64 sum=root->c[1]->c[0]->sum;
			output(sum); puts("");
		}
	}
}

int main()
{
	int num=0;
	while(scanf("%d",&n)!=-1)
	{
		printf("Case #%d:\n",++num);
		deal();
	}
}

 

HDU 4441 Queue Sequence(splay)

标签:blog   http   ar   for   sp   div   c   log   amp   

原文地址:http://www.cnblogs.com/jianglangcaijin/p/4003605.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!