bzoj3343 教主的魔法 分块

Description

---

教主最近学会了一种神奇的魔法，能够使人长高。于是他准备演示给XMYZ信息组每个英雄看。于是N个英雄们又一次聚集在了一起，这次他们排成了一列，被编号为1、2、……、N。
每个人的身高一开始都是不超过1000的正整数。教主的魔法每次可以把闭区间[L, R]（1≤L≤R≤N）内的英雄的身高全部加上一个整数W。（虽然L=R时并不符合区间的书写规范，但我们可以认为是单独增加第L（R）个英雄的身高）
CYZ、光哥和ZJQ等人不信教主的邪，于是他们有时候会问WD闭区间 [L, R] 内有多少英雄身高大于等于C，以验证教主的魔法是否真的有效。
WD巨懒，于是他把这个回答的任务交给了你。

对30%的数据，N≤1000，Q≤1000。
对100%的数据，N≤1000000，Q≤3000，1≤W≤1000，1≤C≤1,000,000,000。

Solution

---

一开始想的是树状数组套主席树，结果发现区间加不好套权值。
看到询问很小n很大可以考虑分块。区间加就对完整的块打标记，两边的暴力改。询问就考虑新开一个数组有序第记录每个块，然后在块里面二分即可
感觉还是不太熟练，要多写题
调了很久发现是对拍的暴力写错了囧

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define max(x,y) ((x)>(y)?(x):(y))
const int N=1000005;
const int block=1000;
int pos[N],h[N],c[N],st[N],ed[N],add[N];
void update(int x) {
    rep(i,st[x],ed[x]) h[i]=c[i];
    std:: sort(h+st[x],h+ed[x]+1);
}
void modify(int l,int r,int x) {
    if (pos[l]==pos[r]) {
        rep(i,l,r) c[i]+=x;
        update(pos[l]);
        return ;
    }
    rep(i,pos[l]+1,pos[r]-1) add[i]+=x;
    rep(i,l,ed[pos[l]]) c[i]+=x; update(pos[l]);
    rep(i,st[pos[r]],r) c[i]+=x; update(pos[r]);
}
int find(int x,int y,int v) {
    int l=x,r=y;
    while (l<=r) {
        int mid=(l+r)>>1;
        if (h[mid]>=v) r=mid-1;
        else l=mid+1;
    }
    return y-l+1;
}
int query(int l,int r,int x) {
    int ret=0;
    if (pos[l]==pos[r]) {
        rep(i,l,r) if (c[i]>=x-add[pos[i]]) ret++;
        return ret;
    }
    rep(i,pos[l]+1,pos[r]-1) {
        ret+=find(st[i],ed[i],x-add[i]);
    }
    rep(i,l,ed[pos[l]]) if (c[i]>=x-add[pos[l]]) ret++;
    rep(i,st[pos[r]],r) if (c[i]>=x-add[pos[r]]) ret++;
    return ret;
}
int main(void) {
    freopen("data.in","r",stdin);
    freopen("myp.out","w",stdout);
    int n,m; scanf("%d%d",&n,&m);
    rep(i,1,n) scanf("%d",&h[i]);
    rep(i,1,n) {
        c[i]=h[i];
        pos[i]=(i-1)/block+1;
        if (!st[pos[i]]) st[pos[i]]=i;
        ed[pos[i]]=max(ed[pos[i]],i);
    }
    rep(i,1,pos[n]) update(i);
    while (m--) {
        char opt[2]; scanf("%s",opt);
        int l,r,x; scanf("%d%d%d",&l,&r,&x);
        if (opt[0]=='A') {
            printf("%d\n", query(l,r,x));
        } else if (opt[0]=='M') {
            modify(l,r,x);
        }
    }
    return 0;
}