分块法 hdu1166 敌兵布阵

当然啦,这题最正规的方法当然是最标准的单点更新的线段树咯


最近发现了分块法这一神器,正尝试水各种数据结构中(线段树,伸展树,主席树....),几乎是万能的(效率只慢了一点)

优势在于

1.占用的空间非常低.

2.非常容易写

3.对于比较复杂的维护和各种数据结构的镶套,能较好的处理


缺点当然就是没有标准的那种数据结构块(一般的数据结构复杂度是O(nlogn),这个的复杂度是O(msqrt(n)),有时候还是会卡时间的)


#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
const int MX=50000+5;

int n,unit;
int A[MX],S[MX];

void build(){
    unit=sqrt(n+0.5);
    memset(S,0,sizeof(S));

    for(int i=1;i<=n;i++){
        scanf("%d",&A[i]);

        int u=sqrt(i+0.5);
        S[u]+=A[i];
    }
}

void update(int x,int d){
    int u=sqrt(x+0.5);
    S[u]+=d;A[x]+=d;
}

int get_sum(int u,int L,int R){
    int l=max(u*u,L),r=min((u+1)*(u+1)-1,R),ret=0;
    for(int i=l;i<=r;i++){
        ret+=A[i];
    }
    return ret;
}

int query(int L,int R){
    int uL=sqrt(L+0.5),uR=sqrt(R+0.5),ret=0;
    for(int i=uL+1;i<=uR-1;i++){
        ret+=S[i];
    }
    ret+=get_sum(uL,L,R);
    if(uL!=uR) ret+=get_sum(uR,L,R);
    return ret;
}

int main(){
    char op[10];
    int T,ansk=0;

    scanf("%d",&T);
    while(T--){
        printf("Case %d:\n",++ansk);
        scanf("%d",&n);

        build();
        while(scanf("%s",op),op[0]!='E'){
            int a,b;
            scanf("%d%d",&a,&b);

            if(op[0]=='A'){
                update(a,b);
            }else if(op[0]=='S'){
                update(a,-b);
            }else{
                printf("%d\n",query(a,b));
            }
        }
    }
    return 0;
}


你可能感兴趣的:(hdu1166)