士兵殺敵（二）

時間限制：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

解法一：

樹狀數組

#include<bits/stdc++.h> using namespace std; int N,M,c[1000005]; int lowbit(int i) {return i&(-i);} void add(int i,int value) {while(i<=N){c[i]+=value;i+=lowbit(i);} } int sum(int i){int sum=0;while(i>0){sum+=c[i];i-=lowbit(i);}return sum; } int main() {ios::sync_with_stdio(false);fill(c,c+1000005,0);cin>>N>>M;int a,b;string op;for(int i=1;i<=N;i++){cin>>a;add(i,a);}while(M--){cin>>op>>a>>b;if(op[0]=='A'){add(a,b);}else{cout<<sum(b)-sum(a-1)<<endl;}}return 0;}

解法二：

線段樹，更新點，查詢區間求和

#include<bits/stdc++.h> using namespace std; const int MAXNODE =1e5+3; const int MAX=MAXNODE<<2; struct NODE{int value;//區間和int left,right; }node[MAXNODE];int father[MAX]; void BuildTree(int i,int left,int right) {node[i].left=left;node[i].right=right;node[i].value=0;if(left==right){father[left]=i;return ;}BuildTree(i<<1,left,(left+right)/2);BuildTree(i<<1|1,(left+left)/2+1,right); } void UpdateTree(int ri) {if(ri==1) return;int fi=ri/2;int a=node[fi<<1].value;int b=node[fi<<1|1].value;node[fi].value=a+b;UpdateTree(ri>>1); }int Sum; void query(int i,int l,int r) {if(node[i].left==l&&node[i].right==r){Sum+=node[i].value;return;} i=i<<1;if(l<=node[i].right){if(r<=node[i].right)query(i,l,r);elsequery(i,l,node[i].right);}i++;if(r>=node[i].left){if(l>=node[i].left)query(i,l,r);else query(i,node[i].left,r);} } int main(){int n,m,g;ios::sync_with_stdio(false);while(cin>>n>>m){BuildTree(1,1,n);for(int i=1;i<=n;i++){cin>>g;node[father[i]].value=g;UpdateTree(father[i]);}while(m--){int a,b;string op;//getchar();cin>>op>>a>>b;if(op[0]=='Q'){Sum=0;query(1,a,b);cout<<Sum<<endl;}else{node[father[a]].value+=b;UpdateTree(father[a]);}}}return 0; }

總結

以上是生活随笔為你收集整理的NYOJ【士兵杀敌（二）】的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。