【权值分块】bzoj1208 [HNOI2004]宠物收养所

不多说。比pb_ds还是要快不少的。

 #include<cstdio>
 #include<algorithm>
 #include<cmath>
 using namespace std;
 #define N 80001
 int sum,sz,num[N],l[],r[],a[N],op[N],en,ma[N],ans,n;
 inline void R(int &x){
     char c=;int f=;
     for(;c<''||c>'';c=getchar())if(c=='-')f=-;
     for(x=;c>=''&&c<='';c=getchar())(x*=)+=(c-'');
     x*=f;
 }
 void makeblock()
 {
     sz=sqrt(en); if(!sz) sz=;
     for(sum=;sum*sz<en;sum++)
       {
         l[sum]=r[sum-]+;
         r[sum]=sum*sz;
         for(int i=l[sum];i<=r[sum];i++) num[i]=sum;
       }
     l[sum]=r[sum-]+;
     r[sum]=en;
     for(int i=l[sum];i<=r[sum];i++) num[i]=sum;
 }
 struct Val_Blocks
 {
     int sumv[],Size;
     bool b[N];
     inline void Insert(const int &x){b[x]=; sumv[num[x]]++; Size++;}
     inline void Delete(const int &x){b[x]=; sumv[num[x]]--; Size--;}
     inline int Next(const int &x)
       {
         if(b[x]) return x;
         for(int i=x+;i<=r[num[x]];i++) if(b[i]) return i;
         for(int i=num[x]+;i<=sum;i++) if(sumv[i])
           for(int j=l[i];;j++)
             if(b[j]) return j;
         return -;
       }
     inline int Pre(const int &x)
       {
         if(b[x]) return x;
         for(int i=x-;i>=l[num[x]];i--) if(b[i]) return i;
         for(int i=num[x]-;i>=;i--) if(sumv[i])
           for(int j=r[i];;j--)
             if(b[j]) return j;
         return -;
       }
 }T[];
 struct Point{int v,p;}t[N];
 bool operator < (const Point &a,const Point &b){return a.v<b.v;}
 int main()
 {
     R(n); for(int i=;i<=n;i++)
       {
           R(op[i]); R(t[i].v);
           t[i].p=i;
       } sort(t+,t+n+);
     ma[a[t[].p]=++en]=t[].v;
     for(int i=;i<=n;i++)
       {
           if(t[i].v!=t[i-].v) en++;
           ma[a[t[i].p]=en]=t[i].v;
       } makeblock();
     for(int i=;i<=n;i++)
       {
           if(!T[op[i]^].Size) T[op[i]].Insert(a[i]);
           else
             {
                 int Up=T[op[i]^].Next(a[i]),Down=T[op[i]^].Pre(a[i]);
                 if(Up==-||(Down!=-&&ma[a[i]]-ma[Down]<=ma[Up]-ma[a[i]]))
                   {
                       ans=(ans+ma[a[i]]-ma[Down])%;
                       T[op[i]^].Delete(Down);
                   }
                 else
                   {
                       ans=(ans+ma[Up]-ma[a[i]])%;
                       T[op[i]^].Delete(Up);
                   }
             }
       } printf("%d\n",ans);
     return ;
 }