Problem Description
A国在海岸线沿直线布置了N个工兵营地
Input
第一行一个整数T,表示有T组数据。
每组数据第一行一个正整数N(N<=50000),表示敌人有N个工兵营地,接下来有N个正整数,第i个正整数ai代表第i个工兵营地里开始时有ai个人(1<=ai<=50)。
接下来每行有一条命令,命令有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 表示结束,这条命令在每组数据最后出现;
每组数据最多有40000条命令
每组数据第一行一个正整数N(N<=50000),表示敌人有N个工兵营地,接下来有N个正整数,第i个正整数ai代表第i个工兵营地里开始时有ai个人(1<=ai<=50)。
接下来每行有一条命令,命令有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 表示结束,这条命令在每组数据最后出现;
每组数据最多有40000条命令
Output
对第i组数据,首先输出“Case i:”和回车,
对于每个Query询问,输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内。
对于每个Query询问,输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内。
Sample Input
1 10 1 2 3 4 5 6 7 8 9 10 Query 1 3 Add 3 6 Query 2 7 Sub 10 2 Add 6 3 Query 3 10 End
Sample Output
Case 1: 6 33 59
题意:对一维数组操作。
题解:利用树状数组
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
char c[6];
int C[50050];
int lowbit(int q){
return q&(-q);
}
void add(int x,int y,int n){
while (x<=n){
C[x]+=y;
x+=lowbit(x);
}
}
int qu(int n){
int s=0;
int i;
while (n>0){
s+=C[n];
n-=lowbit(n);
}
return s;
}
int main(){
int i,x,y;
int t,n,a;
int cas=1;
scanf ("%d",&t);
while (t--){
scanf ("%d",&n);
memset (C,0,sizeof(C));
for (i=1;i<=n;i++){
scanf ("%d",&a);
add(i,a,n);
}
printf ("Case %d:\n",cas++);
while (scanf ("%s",c)&&c[0]!='E'){
scanf ("%d %d",&x,&y);
if (c[0]=='A')
add(x,y,n);
else if (c[0]=='S')
add(x,-y,n);
else{
printf ("%d\n",qu(y)-qu(x-1));
}
}
}
return 0;
}