描述
南將軍手下有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
-
2
-
0
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
代碼:
#include<stdio.h>#include<string.h>int a[1000005],c[1000005];inline int lowbit(int x)//樹狀數組下標的確定{return x&(x^(x-1));}int Sum(int u)//求和{int s=0;while(u>0){s+=c[u];u-=lowbit(u);}return s;}inline void change(int x,int m,int n)//改變值{for(int i=x;i<=n;i+=lowbit(i))c[i]+=m;}int main(){int i,N,M;scanf("%d %d",&N,&M);for(i=1;i<=N;i++){scanf("%d",&a[i]);change(i,a[i],N);}char s[20];int a,b;for(i=0;i<M;i++){scanf("%s",s);scanf("%d %d",&a,&b);if(strcmp(s,"QUERY")==0)printf("%d\n",Sum(b)-Sum(a-1));else if(strcmp(s,"ADD")==0)change(a,b,N);}return 0;}
樹狀數組的應用:http://blog.csdn.net/no_retreats/article/details/7853036