1. 程式人生 > >線段樹專題 A(單點更新)

線段樹專題 A(單點更新)

#include <iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=50000+5;
int num[maxn];
char str[10];
struct Node
{
    int l,r;
    int nsum;
}Tree[maxn*3];
//自上到下建樹,自下到上賦值
void build(int i,int l,int r)
{
    Tree[i].l=l;
    Tree[i].r=r;
    if(l==r)                                //遞迴出口
{ Tree[i].nsum=num[l]; return; } int mid=(l+r)>>1; build(i<<1,l,mid); build((i<<1)+1,mid+1,r); Tree[i].nsum=Tree[i<<1].nsum+Tree[(i<<1)+1].nsum; } //自上到下的更新 void Add(int i,int t,int b) //t為軍營編號,b為變動 { Tree[i].nsum+=b; if
(Tree[i].l==t&&Tree[i].r==t) return; int mid=(Tree[i].l+Tree[i].r)>>1; if(t<=mid) Add(i<<1,t,b); else Add((i<<1)+1,t,b); } //查詢 int query(int i,int l,int r) { if(l==Tree[i].l&&r==Tree[i].r) return Tree[i].nsum; int mid=(Tree[i].l+Tree[i].r)>>1
; if(r<=mid) return query((i<<1),l,r); else if(l>mid) return query((i<<1)+1,l,r); else return query(i<<1,l,mid)+query((i<<1)+1,mid+1,r); } int main() { int t; while(scanf("%d",&t)==1) { int n,kase=0; while(t--) { scanf("%d",&n); for(int i=1;i<=n;i++) scanf("%d",&num[i]); build(1,1,n); printf("Case %d:\n",++kase); while(scanf("%s",&str)==1) { if(str[0]=='E') break; int a,b; scanf("%d%d",&a,&b); if(str[0]=='A') Add(1,a,b); else if(str[0]=='S') Add(1,a,-b); else printf("%d\n",query(1,a,b)); } } } return 0; }