时间限制: 2 Sec 内存限制: 32 MB
题目描述
食堂有N个打饭窗口,现在正到了午饭时间,每个窗口都排了很多的学生,而且每个窗口排队的人数在不断的变化。
现在问你第i个窗口到第j个窗口一共有多少人在排队?
输入
输入的第一行是一个整数T,表示有T组测试数据。
每组输入的第一行是一个正整数N(N<=30000),表示食堂有N个窗口。
接下来一行输入N个正整数,第i个正整数ai表示第i个窗口最开始有ai个人排队。(1<=ai<=50)
接下来每行有一条命令,命令有四种形式:
(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条命令。
输出
对于每组输入,首先输出样例号,占一行。
然后对于每个Query询问,输出一个整数,占一行,表示询问的段中的总人数,这个数保持在int以内。
样例输入
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
样例输出
Case 1:
6
33
59
思路:
单点更新,区间查询。注意数组C[]每轮查询前清零。
code:
#include <iostream>
#include<cstdio>
#include<cstring>
#define lowbit(x) (x&(-x))
#define ll long long
using namespace std;//单点更新,区间查询
int C[30005];
void add(int x,int v){
for(int i=x;i<30005;i+=lowbit(i)){
C[i]+=v;
}
}
int getsum(int x){
int sum=0;
for(int i=x;i>0;i-=lowbit(i))
sum+=C[i];
return sum;
}
int main(int argc, char** argv) {
int T;
scanf("%d",&T);
for(int t=1;t<=T;t++){
memset(C,0,sizeof(C));//每次清零
int N;scanf("%d",&N);
int x;
for(int i=1;i<=N;i++){
scanf("%d",&x);
add(i,x);
}
string q;
printf("Case %d:\n",t);
while(cin>>q&&(q!="End")){
int a,b;cin>>a>>b;
if(q[0]=='Q') printf("%d\n",getsum(b)-getsum(a-1));
else if(q[0]=='A') add(a,b);
else if(q[0]=='S') add(a,-b);
}
}
return 0;
}