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

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

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