hdu1166 敌兵布阵

xiaoxiao2021-02-28  119

题目:给你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; }

转载请注明原文地址: https://www.6miu.com/read-39626.html

最新回复(0)