hdu 1166 敌兵布阵(线段树)

来源：互联网发布：部落冲突巨人升级数据编辑：程序博客网时间：2024/06/11 04:13

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增加addvoid 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;}

0 0