【分块】bzoj3196 Tyvj 1730 二逼平衡树

分块 或 树套树。

在每个块中维护一个有序表，查询时各种二分，全都是分块的经典操作，就不详细说了。

块的大小定为sqrt(n*log2(n))比较快。

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
using namespace std;
int sum,sz,l[400],r[400],num[51001],a[51001],b[51001],x,y,k,n,m,op,tmp[2000];
void makeblock()
{
    memcpy(b,a,sizeof(a));
    sz=sqrt((double)n*log2(n));
    for(sum=1;sum*sz<n;sum++)
      {
        l[sum]=(sum-1)*sz+1;
        r[sum]=sum*sz;
        for(int i=l[sum];i<=r[sum];i++) num[i]=sum;
        sort(b+l[sum],b+r[sum]+1);
      }
    l[sum]=sz*(sum-1)+1; r[sum]=n;
    for(int i=l[sum];i<=r[sum];i++) num[i]=sum;
}
inline void Update()
{
    *lower_bound(b+l[num[x]],b+r[num[x]]+1,a[x])=y;
    sort(b+l[num[x]],b+r[num[x]]+1);
    a[x]=y;
}
inline void Rank()
{
    int ans=0;
    if(num[x]+1>=num[y]) {for(int i=x;i<=y;i++) if(a[i]<k) ans++;}
    else
      {
        for(int i=x;i<=r[num[x]];i++) if(a[i]<k) ans++;
        for(int i=l[num[y]];i<=y;i++) if(a[i]<k) ans++;
        for(int i=num[x]+1;i<num[y];i++) ans+=lower_bound(b+l[i],b+r[i]+1,k)-(b+l[i]);
      }
    printf("%d\n",ans+1);
}
inline void Kth()
{
    if(num[x]+1>=num[y])
      {
        int en=0;
        for(int i=x;i<=y;i++) tmp[++en]=a[i];
        sort(tmp+1,tmp+en+1);
        printf("%d\n",tmp[k]);
      }
    else
      {
        int L=0,R=1000000001;
        while(L!=R)
          {
            int mid=L+R>>1,cnt=0;
            for(int i=x;i<=r[num[x]];i++) if(a[i]<mid) cnt++;
            for(int i=l[num[y]];i<=y;i++) if(a[i]<mid) cnt++; //统计<mid的值数
            for(int i=num[x]+1;i<num[y];i++) cnt+=lower_bound(b+l[i],b+r[i]+1,mid)-(b+l[i]);
            if(cnt>=k) R=mid;
            else L=mid+1;
          }
        printf("%d\n",L-1);
      }
}
inline void Front()
{
    int ans=-2147483647;int* p;
    if(num[x]+1>=num[y]) {for(int i=x;i<=y;i++) if(a[i]>ans && a[i]<k) ans=a[i];}
    else
      {
        for(int i=x;i<=r[num[x]];i++) if(a[i]>ans && a[i]<k) ans=a[i];
        for(int i=l[num[y]];i<=y;i++) if(a[i]>ans && a[i]<k) ans=a[i];
        for(int i=num[x]+1;i<num[y];i++)
          {
            p=lower_bound(b+l[i],b+r[i]+1,k);
            if(p!=b+l[i] && *(p-1)<k && *(p-1)>ans) ans=*(p-1);
          }
      }
    printf("%d\n",ans);
}
inline void Next()
{
    int ans=2147483647;int* p;
    if(num[x]+1>=num[y]) {for(int i=x;i<=y;i++) if(a[i]<ans && a[i]>k) ans=a[i];}
    else
      {
        for(int i=x;i<=r[num[x]];i++) if(a[i]<ans && a[i]>k) ans=a[i];
        for(int i=l[num[y]];i<=y;i++) if(a[i]<ans && a[i]>k) ans=a[i];
        for(int i=num[x]+1;i<num[y];i++)
          {
            p=upper_bound(b+l[i],b+r[i]+1,k);
            if(p!=b+r[i]+1 && *p>k && *p<ans) ans=*p;
          }
      }
    printf("%d\n",ans);
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    makeblock();
    for(int i=1;i<=m;i++)
      {
        scanf("%d%d%d",&op,&x,&y);
        if(op==3) Update();
        else
          {
              scanf("%d",&k);
            if(op==1) Rank();
            else if(op==2) Kth();
            else if(op==4) Front();
            else Next();
          }
      }
    return 0;
}