1 #include <iostream> 2 #include <stdio.h> 3 #include <memory.h> 4 using namespace std; 5 6 int n, a[50005]; 7 char sh[15]; 8 9 int lowbit(int i) //树状数组最巧妙之处:i&(-i) 10 { 11 return i&(-i); 12 } //满足2^k<=t的最大的2^k,其中k为非负整数 13 14 void update(int i, int val) //更新函数 15 { 16 while(i <= n) 17 { 18 a[i] += val; 19 i += lowbit(i); 20 } 21 } 22 23 int sum(int i) //求和函数 24 { 25 int sum = 0; 26 while(i > 0) 27 { 28 sum += a[i]; 29 i -= lowbit(i); 30 } 31 return sum; 32 } 33 34 int main() 35 { 36 int i, val, t, x, y, zz = 1; 37 scanf("%d", &t); 38 while(t--) 39 { 40 memset(a, 0, sizeof(a)); 41 scanf("%d", &n); 42 for(i = 1; i <= n; i++) 43 { 44 scanf("%d", &val); 45 update(i, val); 46 } 47 printf("Case %d:\n", zz++); 48 while(scanf("%s", sh)) 49 { 50 if(sh[0] == 'E') break; 51 scanf("%d %d", &x, &y); 52 if(sh[0] == 'A') update(x, y); 53 else if(sh[0] == 'S') update(x, -y); 54 else printf("%d\n", sum(y)-sum(x-1)); //两段区间和相减 55 } 56 } 57 58 return 0; 59 }
分析过程是简单的,所有的操作加起来一共是三个函数,主要是搞懂他们的意思就一切OK了
然后上面都是树状数组,并不是真正意义上的线段树,要是想好好的搞定的话就用好下面的这个模板就好了
#include <iostream> #include <cstdio> #include <cstring> using namespace std; #define num 50009 int a [num],b[num * 3]; char s[15]; void build(int root,int l,int r){//告诉我根节点和需要处理的区间的范围,我能够给你一颗树 if(l==r) { b[root]=a[l]; return; } int mid = (l + r)/2; build(root * 2, l , mid); build(root * 2 + 1, mid + 1,r); b[root]=b[root*2]+b[root*2+1]; } void update(int root,int l,int r,int p,int val){ int M = ( l + r ) / 2; if(l==r){ b[root]+=val; } //这个是给叶子结点赋值 else{ if(p<=M) update(root*2,l,M,p,val); else update(root*2+1,M+1,r,p,val); b[root] = b[root*2] + b[root*2 + 1]; } } int Query(int L,int R,int root,int l,int r){ //LR是确定一个区间的,是一直都不会变动的 //但是l和r是当前线段树节点左边界和右边界 if(L<=l&&r<=R) return b[root]; int mid = (l+r)/2; if(R<=mid) return Query(L,R,root * 2,l,mid); else if(L>mid) return Query(L,R,root * 2+1,mid+1,r); else{ int ans1=Query(L,R,root * 2,l,mid); int ans2=Query(L,R,root * 2+1,mid+1,r); return ans1+ans2; } } int main() { int t,n,w=1; cin>>t; while(t--){ //memset(a,0,sizeof(a)); //memset(b,0,sizeof(b)); //其次是对树的操作 //最后是将结果返回 scanf("%d",&n); for(int i = 1;i <= n;i++)cin>>a[i]; build(1,1,n);//首先是创建一颗树 //for(int i = 1;i < 3*n;i++)cout<<b[i]<<" "; cout<<endl; printf("Case %d:\n",w++); while(scanf("%s",s)){ if(s[0]=='E')break; int R,L; cin>>L>>R; //将左右两部分完全的输入 if(s[0]=='Q')printf("%d\n",Query(L,R,1,1,n)); else if(s[0]=='A')update(1,1,n,L,R); else if(s[0]=='S')update(1,1,n,L,-R); } } return 0; }
写这个的过程十分的曲折,中途还很不好意思的麻烦了学长一段时间给我详细的挑错,回去后一定得好好的处理好这段代码,争取让自己以后不会再出现这些错误。感谢学长。