hdu1166 敌兵布阵

题目:给你N个数,表示兵营里面的人数,有4种操作

(1) Add i j,i和j为正整数,表示第i个营地增加j个人(j不超过30) 
(2)Sub i j ,i和j为正整数,表示第i个营地减少j个人(j不超过30); 
(3)Query i j ,i和j为正整数,i<=j,表示询问第i到第j个营地的总人数; 
(4)End 表示结束,这条命令在每组数据最后出现; 

思路:块状数组

代码:

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<string>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<list>
#include<numeric>
using namespace std;
#define PI acos(-1.0)
#define LL long long
#define ULL unsigned long long
#define INF 0x3f3f3f3f
#define mm(a,b) memset(a,b,sizeof(a))
#define PP puts("*********************");
template<class T> T f_abs(T a){ return a > 0 ? a : -a; }
template<class T> T gcd(T a, T b){ return b ? gcd(b, a%b) : a; }
template<class T> T lcm(T a,T b){return a/gcd(a,b)*b;}
// 0x3f3f3f3f3f3f3f3f
// 0x3f3f3f3f

const int maxn=5e4+50;
int arr[maxn];//下标从0开始
int sum[maxn];//每个块的后缀和
int siz,tot;//块的大小,数目,第一块为0~siz-1
int n;
void maintain(int pos){//维护pos所在块的sum值
    int lb=pos/siz*siz,rb=min(n,lb+siz)-1;
    if(pos==rb) sum[pos]=arr[pos],pos--;
    for(;pos>=lb;pos--) sum[pos]=sum[pos+1]+arr[pos];
}
void init(){//初始化块状数组
    siz=(int)sqrt(n*1.0);
    tot=n/siz;
    if(n%siz!=0) tot++;
    for(int i=1;i<=tot;i++)
        maintain(min(i*siz,n)-1);
}
void update(int pos,int val){
    arr[pos]+=val;
    maintain(pos);
}
int query(int L,int R){
    int pos1=L/siz,pos2=R/siz;
    int res=0;
    if(pos1==pos2){
        for(int i=L;i<=R;i++)
            res+=arr[i];
        return res;
    }
    for(int i=L;i<pos1*siz+siz;i++) res+=arr[i];
    for(int i=pos1+1;i<pos2;i++) res+=sum[i*siz];
    for(int i=pos2*siz;i<=R;i++) res+=arr[i];
    return res;
}
int main(){

    int T,cas=0,x,y;
    char str[20];
    scanf("%d",&T);
    while(T--){
        scanf("%d",&n);
        for(int i=0;i<n;i++)
            scanf("%d",&arr[i]);
        init();
        printf("Case %d:\n",++cas);
        while(~scanf("%s",str)){
            if(str[0]=='E') break;
            scanf("%d%d",&x,&y);
            if(str[0]=='A'){
                x--;
                update(x,y);
            }
            else if(str[0]=='S'){
                x--;
                update(x,-y);
            }
            else{
                x--;y--;
                int ans=query(x,y);
                printf("%d\n",ans);
            }
        }
    }
    return 0;
}



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值