• 【bzoj1208】宠物收养所——treap


    第二道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
  • 相关阅读:
    mac zsh
    lldb
    c++ 的整形字面值和如何确定常量类型
    Python GTK + 3教程 学习笔记 ——(7)GTK的UI设计理念
    Python GTK + 3教程 学习笔记 ——(6)gnome-tewak-tool代码学习
    c++ 一些资料
    解决github无法加载图片
    Python GTK + 3教程 学习笔记 ——(5)布局 与 glade
    wp rest api 授权方法步骤(使用JWT Authentication插件)
    解决Ionic的ion-slide-box 2条数据渲染问题
  • 原文地址:https://www.cnblogs.com/JKAI/p/6942171.html
Copyright © 2020-2023  润新知