复习线段树。
线段树(Segment Tree)是一种二叉搜索树,它将一个区间划分成一些单元区间,每个单元区间对应线段树中的一个叶结点。
对于线段树中的每一个非叶子节点[a,b],它的左子树表示的区间为[a,(a+b)/2],右子树表示的区间为[(a+b)/2+1,b]。因此线段树是平衡二叉树。叶节点数目为N,即整个线段区间的长度。
使用线段树可以快速的查找某一个节点在若干条线段中出现的次数,时间复杂度为O(logN)。而未优化的空间复杂度为2N,因此有时需要离散化让空间压缩。
http://zh.wikipedia.org/wiki/%E7%BA%BF%E6%AE%B5%E6%A0%91
#include<iostream>
#include<cstdio>
using namespace std;
#define lson i<<1,l,m
#define rson i<<1|1,m+1,r
#define maxn 50005
int sum[maxn<<2];
void PushUp(int i)
{
sum[i]=sum[i<<1]+sum[i<<1|1];
}
void build(int i,int l,int r)
{
if(l==r) {scanf("%d",&sum[i]);return;}
int m=(l+r)>>1;
build(lson);
build(rson);
PushUp(i);
}
void update(int t,int add,int i,int l,int r)
{
if(l==r) {sum[i]+=add;return;}
int m=(l+r)>>1;
if(t<=m) update(t,add,lson);
else update(t,add,rson);
PushUp(i);
}
int query(int L,int R,int i,int l,int r)
{
if(L<=l&&r<=R) return sum[i];
int m=(l+r)>>1;
int ans=0;
if(L<=m) ans+=query(L,R,lson);
if(R>m) ans+=query(L,R,rson);
return ans;
}
int main()
{
int T,n;
scanf("%d",&T);
for(int c=1;c<=T;++c)
{
printf("Case %d:\n",c);
scanf("%d",&n);
build(1,1,n);
char s[10];
while(scanf("%s",s))
{
if(s[0]=='E') break;
int a,b;
scanf("%d%d",&a,&b);
if(s[0]=='Q') printf("%d\n",query(a,b,1,1,n));
else if(s[0]=='A') update(a,b,1,1,n);
else update(a,-b,1,1,n);
}
}
return 0;
}