士兵殺敵二

士兵殺敵（二）

時間限制：1000 ms  |  內存限制：65535 KB

難度：5

描述

南將軍手下有N個士兵，分別編號1到N，這些士兵的殺敵數都是已知的。

小工是南將軍手下的軍師，南將軍經常想知道第m號到第n號士兵的總殺敵數，請你幫助小工來回答南將軍吧。

南將軍的某次詢問之後士兵i可能又殺敵q人，之後南將軍再詢問的時候，需要考慮到新增的殺敵數。

輸入

只有一組測試數據
第一行是兩個整數N,M，其中N表示士兵的個數(1<N<1000000)，M表示指令的條數。(1<M<100000)
隨後的一行是N個整數，ai表示第i號士兵殺敵數目。(0<=ai<=100)
隨後的M行每行是一條指令，這條指令包含了一個字符串和兩個整數，首先是一個字符串，如果是字符串QUERY則表示南將軍進行了查詢操作，後面的兩個整數m,n，表示查詢的起始與終止士兵編號；如果是字符串ADD則後面跟的兩個整數I,A(1<=I<=N,1<=A<=100),表示第I個士兵新增殺敵數爲A.

輸出

對於每次查詢，輸出一個整數R表示第m號士兵到第n號士兵的總殺敵數，每組輸出佔一行

樣例輸入

5 6
1 2 3 4 5
QUERY 1 3
ADD 1 2
QUERY 1 3
ADD 2 3
QUERY 1 2
QUERY 1 5

樣例輸出

6
8
8
20

樹狀數組節省時間的原因：保存的小標是“跳”的， 並不是所有的下標都遍歷一遍再update，比如求c[8]， 只需要求8 一個就好， 存值的時候就更新3個數組

#include<stdio.h>
#define N 1000004

int a[N];
void change(int i, int addn, int sum){
    while(i <= sum){
        a[i] += addn;
        i += (i & (-i));//求下一個下標
    }
}
int get_sum(int lim){
    int s = 0;
    while(lim > 0){
        s += a[lim];
        lim -= (lim & (-lim));
    }
    return s;
}
int main(){
    char str[20];
    int num, com;
    int i;
    int c, d;
    scanf("%d%d", &num, &com);
    for(i = 1; i <= num; i++){
        scanf("%d", &c);
        change(i, c, num);
    }
    while(com--){
        scanf("%s", str);
        scanf("%d%d", &c, &d);
        if(str[0] == 'Q'){
            printf("%d\n", get_sum(d) - get_sum(c-1));    
        }
        else{
            change(c, d, num);
        }        
    }
    return 0;
}