Luogu P2286 [HNOI2004]宠物收养场

https://www.luogu.org/problemnew/show/P2286

题目描述

凡凡开了一间宠物收养场。收养场提供两种服务：收养被主人遗弃的宠物和让新的主人领养这些宠物。

每个领养者都希望领养到自己满意的宠物，凡凡根据领养者的要求通过他自己发明的一个特殊的公式，得出该领养者希望领养的宠物的特点值a（a是一个正整数，a<2^31），而他也给每个处在收养场的宠物一个特点值。这样他就能够很方便的处理整个领养宠物的过程了，宠物收养场总是会有两种情况发生：被遗弃的宠物过多或者是想要收养宠物的人太多，而宠物太少。

被遗弃的宠物过多时，假若到来一个领养者，这个领养者希望领养的宠物的特点值为a，那么它将会领养一只目前未被领养的宠物中特点值最接近a的一只宠物。（任何两只宠物的特点值都不可能是相同的，任何两个领养者的希望领养宠物的特点值也不可能是一样的）如果有两只满足要求的宠物，即存在两只宠物他们的特点值分别为a-b和a+b，那么领养者将会领养特点值为a-b的那只宠物。

收养宠物的人过多，假若到来一只被收养的宠物，那么哪个领养者能够领养它呢？能够领养它的领养者，是那个希望被领养宠物的特点值最接近该宠物特点值的领养者，如果该宠物的特点值为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以后的结果。

输入输出样例

输入样例#1： 复制

5                  
0 2                      
0 4                         
1 3
1 2
1 5

输出样例#1： 复制

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

 

两个treap，一个记主人，另一个记宠物，s1，s2记几个宠物/主人，建议定义一个struct，这样写起来短了几乎1/3

#include<cstdio>
#include<ctime>
#include<algorithm>
const int INF=(1LL<<31)-1;
const int p=1000000;
using namespace std;

const int N=1e6+5;
int n,s1,s2,ans;

struct A
{
	int hea[N],r,tre[N],lc[N],rc[N],cnt;
	
	inline void zig(int &x,int y)
	{
		lc[x]=rc[y],rc[y]=x,x=y;
	}
	
	inline void zag(int &x,int y)
	{
		rc[x]=lc[y],lc[y]=x,x=y;
	}
	
	void ins(int &x,int k)
	{
		if(!x) 
		{
			x=++cnt,hea[x]=rand(),tre[x]=k;
			return;
		}
		if(k<tre[x]) 
		{
			ins(lc[x],k);
			if(hea[lc[x]]<hea[x]) zig(x,lc[x]);
		}else
		{
			ins(rc[x],k);
			if(hea[rc[x]]<hea[x]) zag(x,rc[x]);
		}
	}
	
	int pre(int k)
	{
		int x=r,ret=0;
		while(x)
		{
			if(tre[x]>k) x=lc[x];
				else ret=max(ret,tre[x]),x=rc[x];
		}
		return ret;
	}
	
	int nxt(int k)
	{
		int x=r,ret=INF;
		while(x)
		{
			if(tre[x]<k) x=rc[x];
				else ret=min(ret,tre[x]),x=lc[x];
		}
		return ret;
	}
	
	void del(int &x,int k)
	{
		if(x==0) return;
		if(tre[x]==k)
		{
			if(!lc[x]||!rc[x]) x=lc[x]+rc[x];
				else if(hea[lc[x]]<hea[rc[x]]) zig(x,lc[x]),del(rc[x],k);
					else zag(x,rc[x]),del(lc[x],k);
			return;
		}
		if(tre[x]>k) del(lc[x],k);
			else del(rc[x],k);
	}
}treap1,treap2;

int main()
{
	srand(time(NULL));
	scanf("%d",&n);
	while(n--)
	{
		int t,x; scanf("%d%d",&t,&x);
		if(!t)
		{
			if(s2)
			{
				int a1=treap2.pre(x),a2=treap2.nxt(x);
				if(a1&&x-a1<=a2-x||a2==INF) 
					ans=((x-a1)%p+ans)%p,treap2.del(treap2.r,a1);
				else ans=((a2-x)%p+ans)%p,treap2.del(treap2.r,a2);
				s2--;
			}else treap1.ins(treap1.r,x),s1++;
		}else 
		{
			if(s1)
			{
				int a1=treap1.pre(x),a2=treap1.nxt(x);
				if(a1&&x-a1<=a2-x||a2==INF)
					ans=((x-a1)%p+ans)%p,treap1.del(treap1.r,a1);
				else ans=((a2-x)%p+ans)%p,treap1.del(treap1.r,a2);
				s1--;
			}else treap2.ins(treap2.r,x),s2++;
		}
	}
	printf("%d",ans);
	return 0;
}