分块*2

洛谷

 P3203 [HNOI2010]BOUNCE 弹飞绵羊

#include<stdio.h>
#include<string.h>
#define N 200255
#define M 450
int n,q,a[N];
int stp[N],nxt[N];
void build() {
    memset(nxt,66,sizeof nxt);
    scanf("%d",&n);
    for(int i=0;i<n;i++) scanf("%d",&a[i]);
    for(int i=n-1;i>=0;i--) {
        if(i/M==(i+a[i])/M) stp[i]=stp[i+a[i]]+1, nxt[i]=nxt[i+a[i]];
        else stp[i]=1, nxt[i]=i+a[i];
    }
    scanf("%d",&q);
}
int query(int x) {
    int v=0;
    while(x<n) v+=stp[x], x=nxt[x];
    return v;
}
void update(int x,int p) {
    a[x]=p;
    for(int i=x;i>=x/M*M;i--) {
        if(i/M==(i+a[i])/M) stp[i]=stp[i+a[i]]+1, nxt[i]=nxt[i+a[i]];
        else stp[i]=1, nxt[i]=i+a[i];
    }
}
int main() {
    int o,x,y;
    build();
    while(q--) {
        scanf("%d%d",&o,&x);
        if(o==1) printf("%d\n",query(x));
        else {
            scanf("%d",&y);
            update(x,y);
        }
    }
}

洛谷

 P2801 教主的魔法

#include<stdio.h>
#include<ctype.h>
#include<algorithm>
#define M 1005
using namespace std;
int n,q;
int a[M][M],b[M][M],sz[M],mk[M];
void build() {
    for(int i=0;i<n/M;i++) {
        sz[i]=M;
        for(int j=0;j<M;j++) {
            scanf("%d",&a[i][j]);
            b[i][j]=a[i][j];
        }
        sort(b[i],b[i]+M);
    }
    sz[n/M]=n%M;
    for(int i=0;i<n%M;i++) {
        scanf("%d",&a[n/M][i]);
        b[n/M][i]=a[n/M][i];
    }
    sort(b[n/M],b[n/M]+n%M);
}
void add(int p,int l,int r,int v) {
    for(int i=l;i<=r;i++) a[p][i]+=v;
    for(int i=1;i<=sz[p];i++) b[p][i]=a[p][i];
    sort(b[p]+1,b[p]+sz[p]);
}
int query(int p,int v) {
    int l=0,r=sz[p]-1,mid;
    while(l<r) {
        mid=l+r>>1;
        if(b[p][mid]<v) l=mid+1;
        else r=mid;
    }
    if(b[p][l]<v) return 0;
    return sz[p]-l;
}
int main() {
    int sum,c,l,r,v;
    scanf("%d%d",&n,&q);
    build();
    while(q--) {
        while(isspace(c=getchar()));
        scanf("%d%d%d",&l,&r,&v);
        l--, r--;
        if(c=='M') {
            if(l/M==r/M) add(l/M,l%M,r%M,v);
            else {
                add(l/M,l%M,sz[l/M],v);
                add(r/M,0,r%M,v);
                for(int i=l/M+1;i<r/M;i++) mk[i]+=v;
            }
        }
        else {
            sum=0;
            if(l/M==r/M) {
                for(int i=l%M;i<=r%M;i++) if(a[l/M][i]+mk[l/M]>=v) sum++;
            }
            else {
                for(int i=l%M;i<sz[l/M];i++) if(a[l/M][i]+mk[r/M]>=v) sum++;
                for(int i=0;i<=r%M;i++) if(a[r/M][i]>=v) sum++;
                for(int i=l/M+1;i<r/M;i++) sum+=query(i,v-mk[i]);
            }
            printf("%d\n",sum);
        }
    }
}