【HNOI2004】宠物收养所（treap）

最近，阿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)。
你的任务：你得到了一年当中，领养者和被收养宠物到来收养所的情况，希望你计算所有收养了宠物的领养者的不满意程度的总和。这一年初始时，收养所里面既没有宠物，也没有领养者。
输入
第一行为一个正整数n，n <= 80000，表示一年当中来到收养所的宠物和领养者的总数。
接下来的n行，按到来时间的先后顺序描述了一年当中来到收养所的宠物和领养者的情况。每行有两个正整数a, b，其中a=0表示宠物，a=1表示领养者，b表示宠物的特点值或是领养者希望领养宠物的特点值。（同一时间呆在收养所中的，要么全是宠物，要么全是领养者，这些宠物和领养者的个数不会超过10000个）
输出
有一个正整数，表示一年当中所有收养了宠物的领养者的不满意程度的总和mod 1000000以后的结果。
样例输入 5
0 2
0 4
1 3
1 2
1 5
输出
3
提示
说明：样例的计算方法：(abs(3-2) + abs(2-4)=3，最后一个领养者没有宠物可以领养)

其实用set就可以A了，但为了练手还是写了一个treap

看到了很多麻烦的做法

看到有人什么维护两棵treap的、什么还要取种类的

完全没有必要啊

因为目前treap里面要么全部是人，要么全部是动物，我们只需要记录一个目前treap里面的是什么和它的数量

支持加点删点和查询前驱后驱就可以了

算是一个很简单的板子题了

#include<bits/stdc++.h>
using namespace std;
inline int read(){
 char ch=getchar();
 int res=0,f=1;
 while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar(); }
 while(isdigit(ch)) res=(res<<3)+(res<<1)+(ch^48),ch=getchar();
 return res*f;
}
int n,root=1,tot,ans;
int lc[80105],rc[80105],data[80105],val[80105];
const int inf =1e9+7;
const int mod=1000000;
inline int add(int va)
{
 val[++tot]=va;
 data[tot]=rand();
 return tot;
}
inline void build()
{
 add(-inf);
 add(inf);
 rc[1]=2;
}
inline void zig(int &p)
{
 int q=lc[p];
 lc[p]=rc[q],rc[q]=p,p=q;
}
inline void zag(int &p)
{
 int q=rc[p];
 rc[p]=lc[q],lc[q]=p,p=q;
}
inline void insert(int &p,int va)
{
 if(p==0)
 {
  p=add(va);
  return ;
 }
 if(va==val[p]) return ;
 if(va<val[p])
 {
  insert(lc[p],va);
  if(data[lc[p]]>data[p]) zig(p); 
 }
 else
 {
  insert(rc[p],va);
  if(data[rc[p]]>data[p]) zag(p);
 }
}
inline void delet(int &p,int va)
{
 if(p==0) return;
 if(val[p]==va)
 {
  if(lc[p]||rc[p])
  {
   if(rc[p]==0||val[lc[p]]>val[rc[p]])
   {
    zig(p);
    delet(rc[p],va);
   }
   else
   {
    zag(p);
    delet(lc[p],va);
   }
  }
  else  p=0;
 }
 val[p]>va?delet(lc[p],va):delet(rc[p],va);
}
inline int pre(int va)
{
 int cur=1,p=root;
 while(p)
 {
  if(val[p]==va) return va;
  if(val[p]<va&&val[p]>val[cur]) cur=p;
  p=val[p]>va?lc[p]:rc[p];
 }
 return val[cur];
}
inline int nxt(int va)
{
 int cur=2,p=root;
 while(p)
 {
  if(val[p]==va)return va;
  if(val[p]>va&&val[p]<val[cur]) cur=p;
  p=val[p]>va?lc[p]:rc[p];
 }
 return val[cur];
}
int main(){
 int sum=0,ans=0,flag=0;
 n=read();
 build();
 for(int i=1;i<=n;i++)
 {
  int y=read(),x=read();
  if(!sum) {flag=y,sum++,insert(root,x);}
  else if(flag==y)
  {
   insert(root,x);
   sum++;
  }
  else
  {
   sum--;
   int a=pre(x);
   int b=nxt(x);
   if(a==-inf&&b!=inf)
   {
    ans=(ans+b-x)%mod;
    delet(root,b);
   }
   if(a!=-inf&&b==inf)
   {
    ans=(ans+x-a)%mod;
    delet(root,a);
   }
   if(a!=-inf&&b!=inf)
   {
    if(x-a<=b-x)
    {
     ans+=(x-a);
     delet(root,a);
    }
    else 
    {
     ans+=(b-x);
     delet(root,b);
    }
   }
  }
 }
 cout<<ans<<endl;
 return 0;
}

转载于:https://www.cnblogs.com/stargazer-cyk/p/10366496.html