LOJ10144宠物收养所

HNOI 2004

最近，阿 Q 开了一间宠物收养所。收养所提供两种服务：收养被主人遗弃的宠物和让新的主人领养这些宠物。每个领养者都希望领养到自己满意的宠物，阿 Q 根据领养者的要求通过他自己发明的一个特殊的公式，得出该领养者希望领养的宠物的特点值 a（ a是一个正整数，a<2³¹），而他也给每个处在收养所的宠物一个特点值，这样他就能够很方便的处理整个领养宠物的过程了。

宠物收养所总是会有两种情况发生：被遗弃的宠物过多或者是想要收养宠物的人太多，而宠物太少：

被遗弃的宠物过多时，假若到来一个领养者，这个领养者希望领养的宠物的特点值为，那么它将会领养一只目前未被领养的宠物中特点值最接近 a 的一只宠物。任何两只宠物的特点值都不可能是相同的，任何两个领养者的希望领养宠物的特点值也不可能是一样的。如果有两只满足要求的宠物，即存在两只宠物他们的特点值分别为 a-b 和 a+b ，那么领养者将会领养特点值为 a-b 的那只宠物；
收养宠物的人过多，假若到来一只被收养的宠物，那么哪个领养者能够领养它呢？能够领养它的领养者，是那个希望被领养宠物的特点值最接近该宠物特点值的领养者，如果该宠物的特点值为 a，存在两个领养者他们希望领养宠物的特点值分别为 a-b 和 a+b ，那么特点值为 a-b 的那个领养者将成功领养该宠物。一个领养者领养了一个特点值为 a 的宠物，而它本身希望领养的宠物的特点值为b ，那么这个领养者的不满意程度为 | a-b | 。

你得到了一年当中，领养者和被收养宠物到来收养所的情况，希望你计算所有收养了宠物的领养者的不满意程度的总和。这一年初始时，收养所里面既没有宠物，也没有领养者。

输入格式

第一行为一个正整数n ，表示一年当中来到收养所的宠物和领养者的总数；

接下来的 n 行，按到来时间的先后顺序描述了一年当中来到收养所的宠物和领养者的情况。每行有两个正整数a ，b其中 a=0 表示宠物，a=1 表示领养者，b 表示宠物的特点值或是领养者希望领养宠物的特点值。

同一时间呆在收养所中的，要么全是宠物，要么全是领养者，这些宠物和领养者的个数不会超过 10⁴ 个。

输出格式

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

样例

样例输入

样例输出

样例说明

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

数据范围与提示

对于全部数据，1<=n<=8e4。

____________________________________________________________________

平衡树查找前驱和后继，然后区较小的加入结果。

记得做一个标记，记录当前平衡树中存储的是宠物还是领养者。

fhq_treap

____________________________________________________________________

  1 #include<bits/stdc++.h>

  2 using namespace std;

  3 const int maxn=8e4+10;

  4 const int inf=-1;

  5 int n,cnt,root,bz=-1;

  6 long long ans;

  7 struct node

  8 {

  9     int lc,rc,siz,rd,val;

 10 }tr[maxn];

 11 int newnode(int v)

 12 {

 13     ++cnt;

 14     tr[cnt].val=v;

 15     tr[cnt].lc=tr[cnt].rc=0;

 16     tr[cnt].siz=1;

 17     tr[cnt].rd=rand();

 18     return cnt;

 19 }

 20 void update(int cur)

 21 {

 22     tr[cur].siz=tr[tr[cur].lc].siz+tr[tr[cur].rc].siz+1;

 23 }

 24 int merge(int x,int y)

 25 {

 26     if(x*y==0)return x+y;

 27     if(tr[x].rd<tr[y].rd)

 28     {

 29         tr[x].rc=merge(tr[x].rc,y);

 30         update(x);

 31         return x;

 32     }

 33     else

 34     {

 35         tr[y].lc=merge(x,tr[y].lc);

 36         update(y);

 37         return y;

 38     }

 39 }

 40 void split(int cur,int k,int &x,int &y)

 41 {

 42     if(!cur)x=y=0;

 43     else

 44     {

 45         if(tr[cur].val<=k)

 46         {

 47             x=cur;

 48             split(tr[cur].rc,k,tr[cur].rc,y);

 49         }

 50         else

 51         {

 52             y=cur;

 53             split(tr[cur].lc,k,x,tr[cur].lc);

 54         }

 55         update(cur);

 56     }

 57 }

 58 void insert(int v)

 59 {

 60     int x,y;

 61     split(root,v,x,y);

 62     root=merge(merge(x,newnode(v)),y);

 63 }

 64 int kth(int cur,int c)

 65 {

 66     while(cur)

 67     {

 68         if(tr[tr[cur].lc].siz+1==c)return tr[cur].val;

 69         else if(tr[tr[cur].lc].siz>=c)cur=tr[cur].lc;

 70         else

 71         {

 72             c-=tr[tr[cur].lc].siz+1;

 73             cur=tr[cur].rc;

 74         }

 75     }

 76     return inf;

 77 }

 78 void del(int &cur,int v)

 79 {

 80     int x,y,z;

 81     split(cur,v,x,z);

 82     split(x,v-1,x,y);

 83     y=merge(tr[y].lc,tr[y].rc);

 84     cur=merge(merge(x,y),z);

 85 }

 86 void work(int v)

 87 {

 88     int x,y,z;

 89     split(root,v,x,y);

 90     int a=kth(x,tr[x].siz);

 91     int b=kth(y,1);

 92     if(a==-1 && b!=-1)

 93     {

 94         ans=(ans+b-v)%1000000;

 95         del(y,b);

 96     }

 97     else if(b==-1 && a!=-1)

 98     {

 99         ans=(ans+v-a)%1000000;

100         del(x,a);

101     }

102     else

103     {

104         if(v-a<=b-v)

105         {

106             ans=(ans+v-a)%1000000;

107             del(x,a);

108         }

109         else

110         {

111             ans=(ans+b-v)%1000000;

112             del(y,b);

113         }

114     }

115     root=merge(x,y);

116     if(root==0)bz=-1;

117 }

118 int main()

119 {

120     scanf("%d",&n);

121     for(int op,v,i=0;i<n;++i)

122     {

123         scanf("%d%d",&op,&v);

124         if(bz==-1 || bz==op)

125         {

126             insert(v);

127             bz=op;

128         }

129         else

130         {

131             work(v);

132         }

133     }

134     cout<<ans;

135     return 0;

136 }