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

【bzoj1208】宠物收养所——treap

时间:2017-06-05 00:24:07      阅读:169      评论:0      收藏:0      [点我收藏+]

标签:i++   get   can   else   play   logs   题目   isp   rtu   

第二道treap题,码完之后总是RE……查了两个多小时的错,对拍了三四份代码啊啊啊

刚开始发现if(tree[k].l*tree[k].r==0)k=tree[k].l+tree[k].r;这一句写错了,但改完还是WA或RE……

于是又找了接近两个小时的错啊啊啊,终于发现是左旋操作中的一个t打成了k,枉我之前还一直以为是del或者get_big写错了呢

还是不够细心啊

好了,言归正传。

这道题由于题目说“同一时间呆在收养所中的,要么全是宠物,要么全是领养者”,那么就可以建一棵treap树(用一个变量ty储存当前树类型,即1或0),与当前树类型相同就加入该节点,不同就对比该值与前驱值差和后继值差的绝对值,取较小的一个(若相同则选前驱),接着ans加上该值(记得要取余mod!),然后del前驱或后继,最后得到的ans即为答案。

具体细节看代码。

技术分享
#include<cstdio>
#include<cstdlib>
#include<iostream>
const int mod=1000000;
using namespace std;
int n,root=0,ty,ans=0,size=0,tt,pp,a,b;
struct point
{
    int v,l,rnd,r;
}tree[100002];
void lturn(int &k)
{
    int t=tree[k].r;
    tree[k].r=tree[t].l;
    tree[t].l=k;
    k=t;
}
void rturn(int &k)
{
    int t=tree[k].l;
    tree[k].l=tree[t].r;
    tree[t].r=k;
    k=t;
}
void insert(int &k,int x)
{
    if(k==0)
    {
        size++;k=size;
        tree[k].l=tree[k].r=0;
        tree[k].v=x;tree[k].rnd=rand();
        return;
    }
    if(x>tree[k].v)
    {
        insert(tree[k].r,x);
        if(tree[tree[k].r].rnd<tree[k].rnd)lturn(k);
    }
    else
    {
        insert(tree[k].l,x);
        if(tree[tree[k].l].rnd<tree[k].rnd)rturn(k);
    }
}
void del(int &k,int x)
{
    if(k==0)return;
    if(tree[k].v==x)
    {
        if(tree[k].l*tree[k].r==0)k=tree[k].l+tree[k].r;
        else if(tree[tree[k].l].rnd<tree[tree[k].r].rnd)rturn(k),del(k,x);
        else lturn(k),del(k,x);
    }
    else if(tree[k].v<x)del(tree[k].r,x);
    else del(tree[k].l,x);
}
void get_big(int k,int x)
{
    if(k==0)return;
    if(tree[k].v<=x)
    {
        tt=k;get_big(tree[k].r,x);
    }
    else get_big(tree[k].l,x);
}
void get_small(int k,int x)
{
    if(k==0){return;}
    if(tree[k].v>x)
    {
        pp=k;get_small(tree[k].l,x);
    }
    else get_small(tree[k].r,x);
}
int main()
{
    int num=0;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d %d",&a,&b);
        if(!num){size=0;root=0,insert(root,b),ty=a,num++;}
        else if(a==ty){
            insert(root,b);
            num++;
        }
        else 
        {
            tt=0;get_big(root,b);
            pp=0;get_small(root,b);
            if(!pp||(tt&&tree[tt].v&&b-tree[tt].v<=tree[pp].v-b)){ans=(ans+b-tree[tt].v)%mod;del(root,tree[tt].v);}
            else {ans=(ans+tree[pp].v-b)%mod;del(root,tree[pp].v);}
            num--;
        }
    }
    printf("%d",ans);
    return 0;
}
View Code

 

【bzoj1208】宠物收养所——treap

标签:i++   get   can   else   play   logs   题目   isp   rtu   

原文地址:http://www.cnblogs.com/JKAI/p/6942171.html

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