- 描述
-
南將軍手下有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函數更新數據:
-
void UpDate(int o,int l,int r,int x, int y) { if(l==r) { tree[o].sum+=y; //精確找到節點,sum+=y return ; } int mid=(l+r)>>1; if(x<=mid) UpDate(o<<1,l,mid,x,y); else UpDate(o<<1|1,mid+1,r,x,y); tree[o].sum=tree[o<<1].sum+tree[o<<1|1].sum; }
AC Code: -
#include<cstdio> #include<cstdlib> #include<cstring> #include<cmath> #include<iostream> #include<algorithm> #include<queue> #include<stack> using namespace std; struct Node { int sum,l,r; }tree[1000010<<2]; void Build(int o,int l,int r) { tree[o].l=l; tree[o].r=r; if(l==r) { int a; scanf("%d",&a); tree[o].sum=a; return; } int mid=(l+r)>>1; Build(o<<1,l,mid); Build(o<<1|1,mid+1,r); tree[o].sum=tree[o<<1].sum+tree[o<<1|1].sum; } void UpDate(int o,int l,int r,int x, int y) { if(l==r) { tree[o].sum+=y; return ; } int mid=(l+r)>>1; if(x<=mid) UpDate(o<<1,l,mid,x,y); else UpDate(o<<1|1,mid+1,r,x,y); tree[o].sum=tree[o<<1].sum+tree[o<<1|1].sum; } int QuerySum(int o,int l,int r,int x,int y) { if(l==x&&r==y) return tree[o].sum; int mid=(l+r)>>1; if(y<=mid) return QuerySum(o<<1,l,mid,x,y); else if(x>mid) return QuerySum(o<<1|1,mid+1,r,x,y); else return QuerySum(o<<1,l,mid,x,mid)+QuerySum(o<<1|1,mid+1,r,mid+1,y); } int main() { int N,M; scanf("%d%d",&N,&M); Build(1,1,N); while(M--) { char s[10]; int a,b; scanf("%s%d%d",s,&a,&b); if(s[0]=='A') UpDate(1,1,N,a,b); else printf("%d\n",QuerySum(1,1,N,a,b)); } return 0; }
[NYOJ - 116] 士兵殺敵(二)(線段樹)
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.