codeup 1110 Problem B 区间查询

问题 B: 区间查询

时间限制: 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;
}