HNOI2004 宠物收养所

Posted 2020-10-15

题目：

　　最近，阿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)。【任务描述】你得到了一年当中，领养者和被收养宠物到来收养所的情况，希望你计算所有收养了宠物的领养者的不满意程度的总和。这一年初始时，收养所里面既没有宠物，也没有领养者。

思路：

　　从贪心的角度来思考，与 a 差值最小的一定是 a的前驱或后驱，所以就可以想到用平衡树实现。

　　 根据 ” 不要重复发明轮子 “ 的思想，可以发现STL的set以及可以满足需要的功能了。（其实就是懒，可能还会写一个treap的版本吧。。。）

　　 一开始还因为边界原因写了很多判断，后来参考 hzwer 大佬的做法插入了 inf 和 -inf 。。。。。也根据他的做法省了一个 set 。。。（貌似只有平衡树是自己想的了。。T_T ）

　　 （hzwer的大佬的链接：http://hzwer.com/1311.html ）

代码：

 

#include <bits/stdc++.h>
#define inf 0x7fffffff
#define mod 1000000
using namespace std;
set<int> s;
typedef set<int>::iterator sit;
int main(){
int n,ans=0; 
bool flag=1;
    scanf("%d",&n);
    s.insert(inf);
    s.insert(-inf);
    while(n--){
        int f,a;
        cin>>f>>a;
        if(s.size()==2){
            flag=f;
            s.insert(a);
        }
        else if(flag==f)
            s.insert(a);
        else {
            sit k1,k2;
            k1=s.lower_bound(a);
            k2=k1;
            k2--;
            if(*k1-a>=a-*k2&&*k2!=-inf){
                ans=(ans+a-(*k2))%mod;
                s.erase(k2);
            }
            else {
                ans=(ans+(*k1)-a)%mod;
                s.erase(k1);
            }
        }
    }
    printf("%d\n",ans);
    return 0;
}