士兵殺敵（二）

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

線段樹！

AC碼：

#include<stdio.h> int sum[3000000]; int a,b,t; void Create(int L,int R,int root) {if(L==R){scanf("%d",&sum[root]);return;}int mid=(L+R)/2;Create(L,mid,root*2);Create(mid+1,R,root*2+1);sum[root]=sum[root*2]+sum[root*2+1]; } void Sum(int L,int R,int root) {if(a>R||b<L)return;if(a<=L&&b>=R){t=t+sum[root];return;}int mid=(L+R)/2;Sum(L,mid,root*2);Sum(mid+1,R,root*2+1); } void Add(int L,int R,int root) {if(a<L||a>R)return;if(a==L&&a==R){sum[root]+=b;return;}int mid=(L+R)/2;Add(L,mid,root*2);Add(mid+1,R,root*2+1);sum[root]=sum[root*2]+sum[root*2+1]; } int main() {int n,m,i;char str[7];scanf("%d%d",&n,&m);Create(1,n,1);for(i=1;i<=m;i++){scanf("%s%d%d",str,&a,&b);if(str[0]=='Q'){t=0;Sum(1,n,1);printf("%d\n",t);}else{Add(1,n,1);}}return 0; } 樹狀數組！

AC碼：

#include<stdio.h> int a[1000002]; int n,m; int lowbit(int t) {return t & ( t ^ ( t - 1 ) ) ; } //執行add操作 void add(int i,int v) {while(i<=n){a[i]+=v;i+=lowbit(i);} } //得到前i項和int sum(int i) {int sum=0;while(i>0){sum+=a[i];i-=lowbit(i);}return sum; } void begin() //根據樹狀數組的節點規律，初始化數組 {int t,i,j;for(i=n;i>=1;i--){t=0;for(j=i-lowbit(i)+1;j<=i;j++){t+=a[j];}a[i]=t;} } int main() {int c,b,i,t;char s[6];scanf("%d%d",&n,&m);for(i=1;i<=n;i++)scanf("%d",&a[i]);begin();while(m--){scanf("%s %d %d",s,&c,&b);if(s[0]=='Q'){printf("%d\n", (sum(b) - sum(c-1))); }else{add(c,b);}}return 0; }

總結

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

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