poj 3481

    题目描述:给出一些操作查找序列里的最大最小值。

    还是用SBT做的,其实平衡树很好用,效率也高,代码长一点全当作是锻炼代码能力吧。

 

 

  1. #include "stdio.h"
    #include "string.h"
    #include "stdlib.h"
    const int N=50005;
    const int M=1000005;
    
  2. struct Node
    {
    	int k;
    	int prio;
    	int size;
    	struct Node *l,*r;
    }*root,tree[M];
    
    
    int tot;
    int n,m;
    
    
  3. void init()
    {
    	tot=0;
    	root=NULL;
    }
    
    
  4. int ls(Node *p)
    {
    	return p->l?p->l->size:0;
    }
    
    
  5. int rs(Node *p)
    {
    	return p->r?p->r->size:0;
    }
    
    
  6. void rotate_left(Node* &p)
    {
    	Node *q=p->r;
    	p->r=q->l;
    	q->l=p;
    	q->size=p->size;
    	p->size=ls(p)+rs(p)+1;
    	p=q;
    }
    
    
  7. void rotate_right(Node* &p)
    {
    	Node *q=p->l;
    	p->l=q->r;
    	q->r=p;
    	q->size=p->size;
    	p->size=ls(p)+rs(p)+1;
    	p=q;
    }
    
    
  8. void maintain(Node* &p,bool flag)
    {
    	if(flag)
    	{
    		if(p->r && p->r->r && p->r->r->size>ls(p)) rotate_left(p);
    		else if(p->r && p->r->l && p->r->l->size>ls(p))
    		{
    			rotate_right(p->r);
    			rotate_left(p);
    		}
    	}
    	else
    	{
    		if(p->l && p->l->l && p->l->l->size>rs(p)) rotate_right(p);
    		else if(p->l && p->l->r && p->l->r->size>rs(p))
    		{
    			rotate_left(p->l);
    			rotate_right(p);
    		}
    	}
    }
    
    
  9. void insert(int k,int prio,Node* &p)
    {
    	if(!p)
    	{
    		p=&tree[tot++];
    		p->l=NULL;
    		p->r=NULL;
    		p->size=1;
    		p->k=k;
    		p->prio=prio;
    	}
    	else
    	{
    		p->size++;
    		if(p->prio>prio) insert(k,prio,p->l);
    		else insert(k,prio,p->r);
    		maintain(p,prio>=p->prio);
    	}
    }
    
    
  10. int getmax(int &prio,Node *p)
    {
    	if(!p) return 0;
    	while(p->r) p=p->r;
    	prio=p->prio;
    	return p->k;
    }
    
    
  11. int getmin(int &prio,Node *p)
    {
    	if(!p) return 0;
    	while(p->l) p=p->l;
    	prio=p->prio;
    	return p->k;
    }
    
    
  12. void delet(int prio,Node* &p)
    {
    	if(!p) return ;
    	p->size--;
    	if(prio<p->prio) delet(prio,p->l);
    	else if(prio>p->prio) delet(prio,p->r);
    	else
    	{
    		if(!p->l && !p->r) p=NULL;
    		else if(!p->l) p=p->r;
    		else if(!p->r) p=p->l;
    		else
    		{
    			Node *q;
    			q=p->r;
    			while(q->l) q=q->l;
    			p->prio=q->prio;
    			delet(q->prio,p->r);
    		}
    	}
    }
    
    
    
    
  13. int main()
    {
    	init();
    	while(scanf("%d",&n)==1)
    	{
    		if(!n) break;
    		if(n==1)
    		{
    			int k,p;
    			scanf("%d%d",&k,&p);
    			insert(k,p,root);
    		}
    		else if(n==2)
    		{
    			int k,p;
    			k=getmax(p,root);
    			if(!k) printf("0/n");
    			else
    			{
    				printf("%d/n",k);
    				delet(p,root);
    			}
    		}
    		else
    		{
    			int k,p;
    			k=getmin(p,root);
    			if(!k) printf("0/n");
    			else
    			{
    				printf("%d/n",k);
    				delet(p,root);
    			}
    		}
    	}
    	return 0;
    }
    

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值