本题是一道基本的线段树裸题,是求区间和的问题。
我用了堆式线段树写这道题,还是比较容易理解的,要掌握线段树的应用,就要熟悉递归调用,总之就是不断地递归向子区间,直到进入完全覆盖的区间然后统计区间和即可。只要理解了递归这一核心思想就基本明白了简单线段树如何操作。
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=50001;
const int inf=0x7f7f7f7f;
struct SegmentTree//堆式线段树
{
int l,r,sum;
}tree[N*4];
void BuildTree(int l,int r,int idx)
{
tree[idx].l=l;
tree[idx].r=r;
tree[idx].sum=0;
if (l==r) return;
int mid=(l+r)>>1;
BuildTree(l,mid,idx<<1);//左移一位是乘二
BuildTree(mid+1,r,idx<<1|1);//乘二加一
}
void Insert (int i,int j,int idx)
{
if (tree[idx].l==tree[idx].r)
{
tree[idx].sum+=j;
return;
}
int mid=(tree[idx].l+tree[idx].r)>>1;
if (i<=mid)
Insert (i,j,idx<<1);
else Insert (i,j,idx<<1|1);
tree[idx].sum=tree[idx<<1].sum+tree[idx<<1|1].sum;
}
int Query (int l,int r, int idx)
{
if (tree[idx].l==l&&tree[idx].r==r)
return tree[idx].sum;
int mid=(tree[idx].l+tree[idx].r)>>1;
if (r<=mid)
return Query(l,r,idx<<1);
else if (l>mid)
return Query(l,r,idx<<1|1);
return Query(l,mid,idx<<1)+Query(mid+1,r,idx<<1|1);
}
int main()
{
int ca,T=0;
for (scanf ("%d",&ca);ca--;)
{
cout<<"Case "<<++T<<":\n";//注意Case 后面有个空格
int n;
scanf ("%d",&n);
BuildTree(1,n,1);//先建立一棵空树
for (int i=1;i<=n;i++)
{
int x;
scanf ("%d",&x);
Insert(i,x,1);
}
char s[11];
while (scanf ("%s",s),s[0]!='E')
{
int i,j;
scanf ("%d%d",&i,&j);
if (s[0]=='A')
Insert (i,j,1);
else if (s[0]=='S')
Insert(i,-j,1);
else
printf ("%d\n",Query(i,j,1));
}
}
return 0;
}