[kuangbin带你飞]专题七 线段树 A - 敌兵布阵 （单点修改，区间求和）

#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<iostream>
#define lid (id << 1)
#define rid (id << 1 | 1)
using namespace std;
const int MAXN=50005;
int a[MAXN];
struct Node
{
int l,r;
int sum,Max;
void updata(int x)
{
sum+=x; //增加
//  sum=x; //修改
Max=sum;

}
}tr[MAXN*4];
void push_up(int id)
{
tr[id].sum = tr[lid].sum + tr[rid].sum;
tr[id].Max = max(tr[lid].Max,tr[rid].Max);
}
void build(int id,int l,int r)
{
tr[id].l=l;tr[id].r=r;
if(l==r) tr[id].sum=tr[id].Max=a[l];
else
{
int mid=(l+r)>>1;
build(lid,l,mid);
build(rid|1,mid+1,r);
push_up(id);
}
}
void updata(int id,int x,int v)
{
if(tr[id].l==x&&tr[id].r==x)  tr[id].updata(v);
else{
int mid=(tr[id].l+tr[id].r)>>1;
if(x<=mid)  updata(lid,x,v);
else       updata(rid,x,v);
push_up(id);
}
}
int query_sum(int id,int l,int r)
{
if(l==tr[id].l&&r==tr[id].r) return tr[id].sum;
int mid=(tr[id].l+tr[id].r)>>1;
if(r<=mid)  return query_sum(lid,l,r);
else if(l>mid)   return query_sum(rid,l,r);
else  return query_sum(lid,l,mid)+query_sum(rid,mid+1,r);

}
/*int query_max(int id,int l,int r)
{
if(l==tr[id].l&&r==tr[id].r) return tr[id].Max;
int mid=(tr[id].l+tr[id].r)>>1;
if(r<=mid)  return query_max(lid,l,r);
else if(l>mid)   return query_max(rid,l,r);
else  return max(query_max(lid,l,mid),query_max(rid,mid+1,r));

} */
char str[10];
int main()
{
int T,iCase=1,n;
scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
for(int i=1;i<=n;i++)  scanf("%d",&a[i]);
build(1,1,n);  printf("Case %d:\n",iCase++);
while(scanf("%s",&str))
{
if(strcmp(str,"End")==0) break;
int x,y;
scanf("%d%d",&x,&y);
if(strcmp(str,"Add")==0) updata(1,x,y);
else if(strcmp(str,"Sub")==0)  updata(1,x,-y);
else  printf("%d\n",query_sum(1,x,y));
}
}
return 0;
}