hdu1166-敌兵布阵-分块

把区间分成√n份降低复杂度.

#include<bits/stdc++.h>

#define inf 0x3f3f3f3f

const int maxn=50000;

const int seg=223;

using namespace std;

int t,n,icase,x,y;

char order[maxn/10000+10];

int a[maxn+10];

int sum[maxn+10];

void Add(int x,int y){
    a[x]+=y;
    sum[x/seg+1]+=y;
}

int query(int l,int r){
    int res=0;
    int L=l/seg+1,R=r/seg+1;
    if(L==R||R==L+1){
        for(int i=l;i<=r;i++){
                res+=a[i];
        }
    } else {
       for(int i=l;i<=L*seg-1;i++){
           res+=a[i];
       }
       for(int i=r;i>=(R-1)*seg;i--){
           res+=a[i];
       }
       for(int i=L+1;i<=R-1;i++){
           res+=sum[i];
       }
    }
  return res;
}

void solve(){
    scanf("%d",&n);
    for(int i=0;i<n;i++){
        scanf("%d",&a[i]);
    }
    for(int i=0;i<n;i++){
        sum[i/seg+1]+=a[i];
    }
    printf("Case %d:\n",++icase);
    while(scanf("%s",order)!=EOF){
        if(order[0]=='E'){
           break;
        } else if(order[0]=='A'){
                scanf("%d%d",&x,&y);
                x--;
                Add(x,y);
        } else if(order[0]=='S'){
                scanf("%d%d",&x,&y);
                x--;
                Add(x,-y);
        }
         else if(order[0]=='Q'){
                scanf("%d%d",&x,&y);
                x--,y--;
                int ans=query(x,y);
                printf("%d\n",ans);
        }
    }
}

void init(){
  memset(a,0,sizeof(a));
  memset(sum,0,sizeof(sum));
}

int main()
{
    scanf("%d",&t);
    while(t--){
        init();
        solve();
    }
    return 0;
}