BZOJ1208 宠物收养所

1208: [HNOI2004]宠物收养所

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 11074   Solved: 4460
[ Submit][ Status][ Discuss]

Description

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

Input

第一行为一个正整数n，n<=80000，表示一年当中来到收养所的宠物和领养者的总数。接下来的n行，按到来时间的先后顺序描述了一年当中来到收养所的宠物和领养者的情况。每行有两个正整数a, b，其中a=0表示宠物，a=1表示领养者，b表示宠物的特点值或是领养者希望领养宠物的特点值。（同一时间呆在收养所中的，要么全是宠物，要么全是领养者，这些宠物和领养者的个数不会超过10000个）

Output

仅有一个正整数，表示一年当中所有收养了宠物的领养者的不满意程度的总和mod 1000000以后的结果。

Sample Input

5
0 2
0 4
1 3
1 2
1 5

Sample Output

3
(abs(3-2) + abs(2-4)=3，最后一个领养者没有宠物可以领养)

解析：

    这道题需要注意一些细节地方，否则很容易被坑。。。

    先说说怎么做。我们要注意题目描述中的一句话“同一时间呆在收养所中的，要么全是宠物，要么全是领养者”。这句话蕴含的意思就是不用建两个平衡树而只需要建立一个即可。我们可以先用一个flag来标记当前的平衡树里面存的是宠物还是领养者，用sum表示树中节点数量。

    当sum为0时说明树中已经没有宠物或领养者了，这时就把flag改为当前的种类（宠物/领养者）。否则若当前读进来的类型跟这颗树的类型一样则插入这个数，sum++ 。若读进来的类型跟这颗树的类型不一样则累加ans，sum-- 。

    这道题的坑点：

    1.注意最后要取模。

    2.如果前驱和后继与这个数差值相等，则选择前驱。

代码：

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <string>
#include <cmath>
#include <cctype>
#include <queue>
using namespace std;

const int Max=81000;
const int INF=1e9+7;
const int mod=1000000;
int n,root=1,tot,ans,sum,flag;
struct Treap{int l,r,val,data;};
Treap a[Max],b[Max];

inline int get_int()
{
   int x=0,f=1;
   char c;
   for(c=getchar();(!isdigit(c))&&(c!='-');c=getchar());
   if(c=='-') {f=-1;c=getchar();}
   for(;isdigit(c);c=getchar()) x=(x<<3)+(x<<1)+c-'0';
   return x*f;
}

inline int NEW(int val)
{
   a[++tot].val=val;
   a[tot].data=rand();
   return tot;
}

inline void build()
{
   NEW(-INF);
   NEW(INF);
   a[1].r=2;
}

inline void zig(int &p)
{
   int q=a[p].l;
   a[p].l=a[q].r;a[q].r=p;p=q;
}

inline void zag(int &p)
{
   int q=a[p].r;
   a[p].r=a[q].l;a[q].l=p;p=q;
}

inline void Insert(int &p,int val)
{
   if(p==0)
   {
   	 p=NEW(val);
   	 return;
   }
   if(val==a[p].val) return;
   if(val<a[p].val)
   {
   	 Insert(a[p].l,val);
   	 if(a[a[p].l].data>a[p].data) zig(p);
   }
   else
   {
   	 Insert(a[p].r,val);
   	 if(a[a[p].r].data>a[p].data) zag(p);
   }
}

inline int GetPre(int val)
{
   int ans=1,p=root;
   while(p)
   {
   	 if(a[p].val==val) return val;
   	 if(a[p].val<val&&a[p].val>a[ans].val) ans=p;
   	 a[p].val>val ? p=a[p].l : p=a[p].r;
   }
   return a[ans].val;
}

inline int GetNext(int val)
{
   int ans=2,p=root;
   while(p)
   {
   	 if(a[p].val==val) return val;
   	 if(a[p].val>val&&a[p].val<a[ans].val) ans=p;
   	 a[p].val>val ? p=a[p].l : p=a[p].r;
   }
   return a[ans].val;
}

inline void Remove(int &p,int val)
{
   if(p==0) return;
   if(a[p].val==val)
   {
   	 if(a[p].l || a[p].r)
   	 {
   	   if(a[p].r==0 || a[a[p].l].val > a[a[p].r].val)
   	   {
   	   	 zig(p);
   	   	 Remove(a[p].r,val);
   	   }
   	   else
   	   {
   	   	 zag(p);
   	   	 Remove(a[p].l,val);
   	   }
   	 }
   	 else p=0;
   }
   a[p].val>val ? Remove(a[p].l,val) : Remove(a[p].r,val);
}

int main()
{
   //freopen("lx.in","r",stdin);
   //freopen("lx.out","w",stdout);

   n=get_int();
   build();
   for(int i=1;i<=n;i++)
   {
   	 int y=get_int(),x=get_int();
   	 if(!sum) {flag=y;sum++;Insert(root,x);}
   	 else if(flag==y)
	 {
	   Insert(root,x);
	   sum++;
     }
   	 else
   	 {
   	   sum--;
	   int Pre=GetPre(x),Next=GetNext(x);
	   if(Pre==-INF&&Next!=INF)
	   {
	     ans=(ans+Next-x)%mod;
	     Remove(root,Next);
	   }
	   if(Pre!=-INF&&Next==INF)
	   {
	     ans=(ans+x-Pre)%mod;
	     Remove(root,Pre);
	   }
	   if(Pre!=-INF&&Next!=INF)
	   {
	     if(x-Pre <= Next-x) {ans+=x-Pre;Remove(root,Pre);}
		 else {ans=(ans+Next-x)%mod;Remove(root,Next);}
	   }
     }
   }

   cout<<ans<<"\n";
   return 0;
}