hdu1166 敌兵布阵

来源：互联网发布：淘宝1元包邮怎么赚钱编辑：程序博客网时间：2024/06/04 15:04

题目：给你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// 0x3f3f3f3fconst int maxn=5e4+50;int arr[maxn];//下标从0开始int sum[maxn];//每个块的后缀和int siz,tot;//块的大小，数目，第一块为0~siz-1int 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;}

阅读全文

0 0