HDU 1166 敌兵布阵

数状数组 模板题

/*
(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 表示结束，这条命令在每组数据最后出现;
当然线段树也可以解决，不过没树状数组方便 哈哈
*/
#include <iostream>
#include <stdlib.h>
#include <string.h>
using namespace std;

const int MAXN = 50001;

int camp[MAXN], n;
int a[MAXN];

inline int lowbit(int t){
    return t & (-t);
}
void add(int i, int v){
    while(i <= n){
        a[i] += v;
        i += lowbit(i);
    }
}
int sum(int i){
    int s = 0;
    while(i > 0){
        s += a[i];
        i -= lowbit(i);
    }
    return s;
}

int main(){
    int t, cas = 0, i, j;
    char cmd[10];
    bool end = false;
    cin >> t;
    while( t-- ){
        cout << "Case " << ++cas << ":" << endl;
        memset(a, 0, sizeof(a));
        end = false;
        cin >> n;
        for(int i = 1; i <= n; ++i){
            cin >> camp[i];
            add(i, camp[i]);
        }
        while(cin >> cmd)
        {
            switch(cmd[0])
            {
            case 'A':
                cin >> i >> j;
                add(i, j);
                break;
            case 'S':
                cin >> i >> j;
                add(i, -j);    
                break;
            case 'Q':
                cin >> i >> j;
                cout << sum(j) - sum(i - 1) << endl;
                break;
            default:
                end = true;
                break;
            }
            if(end)
                break;
        }
    }
    return 0;
}