線段樹專題 A(單點更新)
阿新 • • 發佈:2018-12-24
#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;
}