http://acm.hdu.edu.cn/showproblem.php?pid=1166
人生第一道线段树题目,一直觉得线段树好高端。。该题应该是线段树最简单的问题,单点增减,包括建树,查询,更新。
#include <stdio.h>
#include <string.h>
const int maxn = 50005;
//线段树节点定义,表示某一线段的左端点,右端点,以及该区间的人数之和
struct line
{
int l;
int r;
int value;
}tree[4*maxn];
int n,a[maxn];
//建树,v是当前线段树的编号,l,r表示对应线段树的左右端点
void build(int v, int l, int r)
{
tree[v].l = l;
tree[v].r = r;
if(l == r)
{
tree[v].value = a[r];
return;
}
int mid = (l+r)>>1;
build(v*2,l,mid);
build(v*2+1,mid+1,r);
tree[v].value = tree[v*2].value + tree[v*2+1].value;
}
//查询左右端点是l,r的线段树的value,v是待查询节点
int query(int v, int l, int r)
{
if(tree[v].l == l && tree[v].r == r)
{
return tree[v].value;
}
int mid = (tree[v].l+tree[v].r)>>1;
if(r <= mid)
return query(v*2,l,r);
else if(l > mid)
return query(v*2+1,l,r);
else return (query(v*2,l,mid) + query(v*2+1,mid+1,r));
}
//单点更新,更新端点是pos的节点,value增加add
void update(int v,int pos,int add)
{
if(tree[v].l == tree[v].r && tree[v].r == pos)
{
tree[v].value += add;
return;
}
int mid = (tree[v].l + tree[v].r)>>1;
if(pos <= mid)
update(v*2,pos,add);
else update(v*2+1,pos,add);
tree[v].value = tree[v*2].value + tree[v*2+1].value;//儿子节点值改变,父亲节点值也要改变
}
int main()
{
int test;
char str[10];
scanf("%d",&test);
for(int item = 1; item <= test; item++)
{
scanf("%d",&n);
for(int i = 1; i <= n; i++)
scanf("%d",&a[i]);
build(1,1,n);
printf("Case %d:\n",item);
int l,r,ans;
int pos,add;
while(scanf("%s",str) && str[0] != 'E')
{
if(str[0] == 'Q')
{
scanf("%d %d",&l,&r);
ans = query(1,l,r);
printf("%d\n",ans);
}
else if(str[0] == 'A')
{
scanf("%d %d",&pos,&add);
update(1,pos,add);
}
else if(str[0] == 'S')
{
scanf("%d %d",&pos,&add);
update(1,pos,-add);
}
}
}
return 0;
}