题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1166
第一次做线段树,帆哥的一句话,我记下来了,其实,线段树就是一种处理数据查询和更新的手段。
然后,我的代码风格,是网上的大牛们的辛苦总结,我就套用了。这里,我还是简单说一下线段树,说的不好,主要方便自己复习。
线段树,3个步骤,建树,查询,更新,
建树:到底部就是a[]数组的值,建立左右子树后,向上推根,根为左右子树的值
更新:类似建树,二分,找到单点所在区间,更新该区间,记得上一个区间也要变化,搜到底部后,加起来add。
查询:查询区间可能横跨,ret要更新,递归搜到该区间大于新的区间,就是sum[rt]
#include <stdio.h> #include <string.h> #define lson l,m,rt<<1 #define rson m+1,r,rt<<1 | 1 const int maxn = 55555; int sum[maxn<<2]; void PushUP(int rt) { sum[rt]=sum[rt<<1]+sum[rt<<1|1]; } void build(int l,int r,int rt) { if(l==r) { scanf("%d",&sum[rt]); return; } int m = (l+r)>>1; build(lson); build(rson); PushUP(rt); }
//单点更新 void update(int p,int add,int l,int r,int rt) { if(l==r) { sum[rt] +=add; return ; } int m=(l+r)>>1; if(p<=m) update(p,add,lson); else update(p,add,rson); PushUP(rt); }
//区间总和 int query(int L,int R,int l,int r,int rt) { if(L<=l&&r<=R) return sum[rt]; int m = (l+r)>>1; int ret = 0; if(L<=m) ret+=query(L,R,lson); if(R>m) ret+=query(L,R,rson); return ret; } int main() { int T,n; scanf("%d",&T); for(int cas = 1;cas<= T;cas++) { printf("Case %d:\n",cas); scanf("%d",&n); build(1,n,1); char ch[10]; while(scanf("%s",ch),strcmp(ch,"End")) { int a,b; scanf("%d%d",&a,&b); if(!strcmp(ch,"Query")) printf("%d\n",query(a,b,1,n,1)); else if(!strcmp(ch,"Add")) update(a,b,1,n,1); else update(a,-b,1,n,1); } } return 0; }