HNOI2004宠物收养场

题目描述

凡凡开了一间宠物收养场。收养场提供两种服务：收养被主人遗弃的宠物和让新的主人领养这些宠物。
每个领养者都希望领养到自己满意的宠物，凡凡根据领养者的要求通过他自己发明的一个特殊的公式，得出该领养者希望领养的宠物的特点值$a$（$a$是一个正整数，$a<2^{31}$），而他也给每个处在收养场的宠物一个特点值。这样他就能够很方便的处理整个领养宠物的过程了，宠物收养场总是会有两种情况发生：被遗弃的宠物过多或者是想要收养宠物的人太多，而宠物太少。
被遗弃的宠物过多时，假若到来一个领养者，这个领养者希望领养的宠物的特点值为$a$，那么它将会领养一只目前未被领养的宠物中特点值最接近$a$的一只宠物。（任何两只宠物的特点值都不可能是相同的，任何两个领养者的希望领养宠物的特点值也不可能是一样的）如果有两只满足要求的宠物，即存在两只宠物他们的特点值分别为$a-b$和$a+b$，那么领养者将会领养特点值为$a-b$的那只宠物。
收养宠物的人过多，假若到来一只被收养的宠物，那么哪个领养者能够领养它呢？能够领养它的领养者，是那个希望被领养宠物的特点值最接近该宠物特点值的领养者，如果该宠物的特点值为$a$，存在两个领养者他们希望领养宠物的特点值分别为$a-b$和$a+b$，那么特点值为$a-b$的那个领养者将成功领养该宠物。
一个领养者领养了一个特点值为a的宠物，而它本身希望领养的宠物的特点值为b，那么这个领养者的不满意程度为$|a-b|$。
你得到了一年当中，领养者和被收养宠物到来收养所的情况，请你计算所有收养了宠物的领养者的不满意程度的总和。这一年初始时，收养所里面既没有宠物，也没有领养者。

输入输出格式

输入格式：

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

输出格式：

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

输入样例

5
0 2
0 4
1 3
1 2
1 5

输出样例

3

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

#include<cstdio>
#include<cstring>
using namespace std;
int root;
struct trnode
{
    int d,n,c,f,son[2];//d为值，f为父亲的编号，c为控制的节点个数，n为同值的节点个数
}tr[110000];int len;
void update(int x)//更新x所控制的节点数 
{
    int lc=tr[x].son[0],rc=tr[x].son[1];
    tr[x].c=tr[lc].c+ tr[rc].c+ tr[x].n;
}
void add(int d,int f)//添加值为d的点，认f为父亲，同时，f也认它为孩子
{
    len++;
    tr[len].d=d;tr[len].n=1; tr[len].c=1;
    tr[len].f=f;if(d<tr[f].d)tr[f].son[0]=len;elsetr[f].son[1]=len; 
    tr[len].son[0]=tr[len].son[1]=0;
}
void rotate(int x,int w)//左旋(x,0)或者右旋(x,1)
{
    int f= tr[x].f,ff=tr[f].f;//x在旋转之前，要确定x的父亲f和爷爷ff 
    //下来建立关系 
    int r,R;//r表示儿辈，R表示父辈 
    //有四个角色：我x，我的儿子，我的父亲，我的爷爷 
    r=tr[x].son[w];R=f;//x的儿子->准备当新儿子 
    tr[R].son[1-w]=r ;
    if(r)tr[r].f=R;//如果x根本就没有这个儿子就不用确定这个儿子的父亲了
    r=x;R=ff;//x->准备当新儿子
    if( tr[R].son[0]==f) tr[R].son[0]=r;else  tr[R].son[1]=r;
    tr[r].f=R;
    r=f;R=x;//x的父亲->准备当新儿子
    tr[R].son[w]=r;
    tr[r].f=R;
    update(f);//先更新处于下层的点f
    update(x);//再更新上层的x
}
void splay(int x,int rt)//该函数功能是为了让x成为rt的孩子（左或右都行）
{
    while(tr[x].f!=rt)//如果rt是x的父亲，则什么都不用做，否则x就要不断向上旋转 
    {
        int f=tr[x].f,ff=tr[f].f;//准备x的父亲和爷爷
        if( ff==rt)//如果x的爷爷是rt，那么x只需要旋转一次（相当于跳一层）
        {
            if( tr[f].son[0]==x) rotate(x,1) ; else rotate(x,0);
        } 
        else
        {
                 if(tr[ff].son[0]==f&&tr[f].son[0]==x)rotate(f,1),rotate(x,1);
            else if(tr[ff].son[1]==f&&tr[f].son[1]==x)rotate(f,0),rotate(x,0);
            else if(tr[ff].son[0]==f&&tr[f].son[1]==x)rotate(x,0),rotate(x,1);
            else if(tr[ff].son[1]==f&&tr[f].son[0]==x)rotate(x,1),rotate(x,0);
        }
    }
    if(rt==0) root=x;
}
int findip(int d)//找值为d的节点的地址，补充：如果不存在d，有可能是接近d的（或大或小）
{
    int x=root;
    while(tr[x].d^d)
    {
        if(d<tr[x].d)
        {
            if( tr[x].son[0]==0)break;
            else x=tr[x].son[0];
        }
        else//if( tr[x].d<d)
        {
            if(tr[x].son[1]==0)break;
            else x=tr[x].son[1];
        }
    }
    return x;
}
void ins(int d)//插入数值为d的一个节点
{
    if(root==0){add(d,0);root=len;return;}
    int x=findip(d);
    if( tr[x].d==d) 
    {
        tr[x].n++;
        update(x);
        splay(x,0);
    }
    else
    { 
        add(d,x);
        update(x);
        splay(len,0);
    }
}
void del(int d)//删除数值为d的一个节点
{
    int x=findip(d);splay(x,0);//找人，并且让y 当树根
    if( tr[x].n>1){tr[x].n--;update(x);return;}// 多重身份，就不用删点
         if(tr[x].son[0]==0&&tr[x].son[1]==0)root=len=0;//四种情况
    else if(tr[x].son[0]==0&&tr[x].son[1]!=0)root=tr[x].son[1],tr[root].f=0;
    else if(tr[x].son[0]!=0&&tr[x].son[1]==0)root=tr[x].son[0],tr[root].f=0;
    else//if( tr[x].son[0]&& tr[x].son[1])
    {
        int p=tr[x].son[0];while(tr[p].son[1])p=tr[p].son[1];splay(p,x);
        int r=tr[x].son[1],R=p;
        tr[R].son[1]=r;tr[r].f=R;
        root=R;tr[root].f=0;
        update(R);
    }
}
int findqianqu(int d)//找小于等于d且最大的数
{
	int x=findip(d);splay(x,0);
	if(d<tr[x].d&&tr[x].son[0])
	{
		x=tr[x].son[0];
		while(tr[x].son[1])x=tr[x].son[1];
	}
	return x;
}
int findhouji(int d)//找大于等于d且最小的数
{
	int x=findip(d);splay(x,0);
	if(tr[x].d<d&&tr[x].son[1])
	{
		x=tr[x].son[1];
		while(tr[x].son[0])x=tr[x].son[0];
	}
	return x;
}
#define myabs(x) ((x)>0?(x):-(x))
#define mymin(x,y) ((x)<(y)?(x):(y))
long long ans=0;//不开long long会爆
int c[2]={0};
int main()
{
	int n;scanf("%d",&n);
	for(int i=1;i<=n;++i)
	{
		int cz,x;scanf("%d%d",&cz,&x);
		if(c[cz^1]==0/*c[1-cz]==0*/)++c[cz],ins(x);
		//如果当时来了个人没有宠物或来了个宠物没有人，就讲ta加入伸展树中
		else//if(c[cz])
		{
			int a=findqianqu(x),b=findhouji(x);
			//a为大于等于d且最小的数，b为大于等于d且最小的数
			if(tr[a].d>x)ans+=myabs(x-tr[a].d),del(tr[a].d);
			//如果找的小于等于d的数仍大于d，就说明没有小于等于d的数
			else if(x>tr[b].d)ans+=myabs(x-tr[b].d),del(tr[b].d);
			//反之亦然
			else
			{
				ans=(ans+mymin(myabs(x-tr[a].d),myabs(x-tr[b].d)));
				//ans加上领养者的不满意程度|a-x|和|b-x|最小的那个
				del(myabs(x-tr[a].d)<=myabs(x-tr[b].d)?tr[a].d:tr[b].d);
				//删除被领养的宠物或已领养宠物的人
			}
			--c[cz^1];//减去这个
		}
	}
	printf("%lld\n",ans%1000000);//按题意模取
}