NY–116 — 士兵殺敵（二） [樹狀數組是也]

 

士兵殺敵（二） 時間限制：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

 

Code:

 

這裡求和的時候考慮情況太多了點....有時候專家層級的指導思想還是沒好處的...

簡簡單單多好~

 

#include"stdio.h"#include"stdlib.h"int n,m;int sum[1000000];int ans;int minpower(int x)//最小冪 {return x&(-x);}void update(int i,int add)//有一個數改變,就把之後含有該項的和全部改變 {while(i<=n){sum[i] += add;i += minpower(i);}}int getsum(int a,int b,int ans)//求區間a~b計程車兵共殺了多少敵人 {//這裡求和想的太麻煩了,考慮情況太多了...其實就把b之前的一直加到1,再減去a之前加到1的就可以了.. int j = minpower(b);int sub = 0;ans += sum[b];//printf("ans:%d b:%d\n",ans,b);if((b-j+1)>a)//沒有加到a的話,接著加上剩餘的 {getsum(a,b-j,ans);}else if((b-j+1)==a) return ans;//如果正好加到a,返回 else{return (ans - getsum(b-j+1,a-1,sub));//加到a前面的話減去多加的值 }} int main(){char order[6];int i,j,k,a,b;scanf("%d%d",&n,&m);sum[0] = 0;for(i=1;i<=n;i++){scanf("%d",&sum[i]);j = minpower(i);for(k=i-1;k>=i-j+1;)//邊輸入邊求和{sum[i] += sum[k];k = k - minpower(k);}}//printf("--------\n");//for(i=1;i<=n;i++)//printf("%d ",sum[i]);//printf("\n--------\n");for(j=1;j<=m;j++){ans = 0;getchar();scanf("%s%d%d",order,&a,&b);switch(order[0]){case 'q':case 'Q'://printf("-----\n");printf("%d",getsum(a,b,ans));if(j!=m) printf("\n");break;case 'A':case 'a':update(a,b);//printf("--------\n");//for(i=1;i<=n;i++)//printf("%d ",sum[i]);//printf("\n--------\n");break;}}return 0;}