首页 > 代码库 > [bzoj1208][HNOI2004]宠物收养所

[bzoj1208][HNOI2004]宠物收养所

题目好长啊直接copy算了

最近,阿Q开了一间宠物收养所。收养所提供两种服务:收养被主人遗弃的宠物和让新的主人领养这些宠物。每个领养者都希望领养到自己满意的宠物,阿Q根据领养者的要求通过他自己发明的一个特殊的公式,得出该领养者希望领养的宠物的特点值a(a是一个正整数,a<2^31),而他也给每个处在收养所的宠物一个特点值。这样他就能够很方便的处理整个领养宠物的过程了,宠物收养所总是会有两种情况发生:被遗弃的宠物过多或者是想要收养宠物的人太多,而宠物太少。 1. 被遗弃的宠物过多时,假若到来一个领养者,这个领养者希望领养的宠物的特点值为a,那么它将会领养一只目前未被领养的宠物中特点值最接近a的一只宠物。(任何两只宠物的特点值都不可能是相同的,任何两个领养者的希望领养宠物的特点值也不可能是一样的)如果有两只满足要求的宠物,即存在两只宠物他们的特点值分别为a-b和a+b,那么领养者将会领养特点值为a-b的那只宠物。 2. 收养宠物的人过多,假若到来一只被收养的宠物,那么哪个领养者能够领养它呢?能够领养它的领养者,是那个希望被领养宠物的特点值最接近该宠物特点值的领养者,如果该宠物的特点值为a,存在两个领养者他们希望领养宠物的特点值分别为a-b和a+b,那么特点值为a-b的那个领养者将成功领养该宠物。 一个领养者领养了一个特点值为a的宠物,而它本身希望领养的宠物的特点值为b,那么这个领养者的不满意程度为abs(a-b)。【任务描述】你得到了一年当中,领养者和被收养宠物到来收养所的情况,希望你计算所有收养了宠物的领养者的不满意程度的总和。这一年初始时,收养所里面既没有宠物,也没有领养者。

set练习题或者平衡树练习题233

set用的不是很熟练,干脆就插两个端点算了。

#include<iostream>
#include<cstdio>
#include<set>
#include<algorithm>
#define itr set<int>::iterator
#define mod 1000000
#define INF 2000000000
using namespace std;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<0||ch>9){if(ch==-) f=-1;ch=getchar();}
    while(ch>=0&&ch<=9){x=x*10+ch-0; ch=getchar();}
    return x*f;
}
multiset<int> s;
int n,kind,ans;
inline int abs(int x){return x<0?-x:x;}

int main()
{
    n=read();s.insert(INF);s.insert(-INF);
    for(int i=1;i<=n;i++)
    {
        int k=read(),x=read();
        if(s.size()==2)s.insert(x),kind=k;
        else if(kind==k)s.insert(x);
        else
        {
            itr l,r=s.lower_bound(x);
            if(*r==INF){l=--r;ans=(ans+x-*l)%mod;s.erase(l);}
            else
            {
                l=--s.lower_bound(x);
                if(*l!=-INF&&(*r-x>=x-*l)){ans=(ans+x-*l)%mod;s.erase(l);}
                else {ans=(ans+*r-x)%mod;s.erase(r);}
            }
        }
    }
    cout<<ans;
    return 0;
}

 splay练习 居然跑的比set快  换一个常数小的平衡树说不定就卡到首页了呢

#include<iostream>
#include<cstdio>
#include<set>
#include<algorithm>
#define mod 1000000
#define INF 2000000000
using namespace std;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<0||ch>9){if(ch==-) f=-1;ch=getchar();}
    while(ch>=0&&ch<=9){x=x*10+ch-0; ch=getchar();}
    return x*f;
}
int n,cnt=0,rt,kind,num=0,ans=0;
int s[80005];
int c[80005][2],fa[80005],size[80005];

inline int abs(int x){return x<0?-x:x;}

void update(int x)
{
    size[x]=size[c[x][0]]+size[c[x][1]]+1;
}

void rotate(int x,int&k)
{
    int y=fa[x],z=fa[y],l=c[y][1]==x,r=l^1;
    if(y==k)k=x;
    else c[z][c[z][1]==y]=x;
    fa[x]=z;fa[y]=x;fa[c[x][r]]=y;
    c[y][l]=c[x][r];c[x][r]=y;
    update(y);update(x);
}

void splay(int x,int&k)
{
    while(x!=k)
    {
        int y=fa[x],z=fa[y];
        if(y!=k)
        {
            if(c[z][1]==y^c[y][1]==x)rotate(x,k);
            else rotate(y,k);
        }
        rotate(x,k);
    }
}

void ins(int&x,int num,int last)
{
    if(!x){x=++cnt;s[x]=num;size[x]=1;fa[x]=last;splay(x,rt);return;}
    if(num<s[x])ins(c[x][0],num,x);
    else ins(c[x][1],num,x);
}

int find2(int k,int x)
{
    if(!k)return 0;
    if(s[k]==x)return k;int q;
    if(s[k]>x)return !(q=find2(c[k][0],x))?k:q;
    else return find2(c[k][1],x);
}

int find1(int k,int x)
{
    if(!k)return 0;
    int q;
    if(s[k]<x)return !(q=find1(c[k][1],x))?k:q;
    else return find1(c[k][0],x);
}

void del(int x)
{
    splay(find1(rt,s[x]),rt);splay(find2(rt,s[x]+1),c[rt][1]);
    c[c[rt][1]][0]=0;update(c[rt][1]);update(rt);
}

int main()
{
    n=read();ins(rt,-INF,0);ins(rt,INF,0);
    for(int i=1;i<=n;i++)
    {
        int k=read(),x=read();
        if(num==0)kind=k,ins(rt,x,0),num++;
        else if(kind==k) ins(rt,x,0),num++;
        else
        {
            int l=find1(rt,x),r=find2(rt,x);splay(l,rt);splay(r,rt);
            if(l<=2){ans=(ans+abs(s[r]-x))%mod;del(r);}
            else if(r<=2||s[r]-x>=x-s[l]){ans=(ans+abs(x-s[l]))%mod;del(l);}
            else {ans=(ans+abs(s[r]-x))%mod;del(r);}
            num--;
        }
    }
    cout<<ans<<endl;
    return 0;
}

 

[bzoj1208][HNOI2004]宠物收养所