【bzoj1208】【HNOI2004】【宠物收养所】【平衡树】【set】

题目大意

有一些宠物和领养人，都有一个特征值，有一些事件加入宠物或领养人，将找到最近的领养人或宠物配对并产生特征值差的不满值，求不满值和。

解题思路

用数据结构维护，查找最近的点。

code

#include<set>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define LF double
#define LL long long
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
using namespace std;
int const maxn=30000,inf=2147483647,mod=1000000;
int n,ans;
multiset<int>s;
void qury(int v){
    int l=*(--s.lower_bound(v)),r=*s.lower_bound(v);
    if((v-l<=r-v)&&(l!=-inf)){
        ans=(ans+v-l)%mod;
        s.erase(--s.lower_bound(v));
    }else{
        ans=(ans+r-v)%mod;
        s.erase(s.lower_bound(v));
    }
}
int main(){
    freopen("d.in","r",stdin);
    freopen("d.out","w",stdout);
    scanf("%d",&n);int tag=0;s.insert(-inf);s.insert(inf);
    fo(i,1,n){
        int a,b;scanf("%d%d",&a,&b);
        if(!a){
            if(tag>=0)s.insert(b);
            else qury(b);
            tag++;
        }else{
            if(tag<=0)s.insert(b);
            else qury(b);
            tag--;
        }
    }
    printf("%d",ans);
    return 0;
}