1 #include<stdio.h> 2 /求树中o 节点位置对应区间的数和 (建立线段树) 3 void build(int o,int l,int r) 4 { 5 if(l==r){ 6 sum[o]=a[l];/定义数组sum保 存信息,a[i]表示数组原来的值 7 return ; 8 }/说明为叶子节点,那么sum [o]就是a [i]的值 9 int mid=(l+r)>>1; //(l+r)/2 10 build(o<<1 ,l, mid); 11 build(o<<1 | 1;mid+1,r); 12 sum[o]=sum[o<<1] + sum[o<<1 | 1]; 13 14 /若求区间最值 15 sum[o]=max(sum[2*o],sum[2*o+1]); 16 } 17 18 /求ql到qr区间的和 (查找) 19 从根开始往下找,直到子节点完全在范围内,反馈信息; 20 21 int ql,qr; 22 23 int query(int o,int l,int r) 24 { 25 if(ql<=l && qr>=r)/节点区间在要找的范围内 26 { 27 return sum [o]; 28 } 29 int mid=(l+r) >> 1; 30 int ans = 0;/定义子区间要返回的值 31 if(ql <= mid)/若左儿子与该要求的区间有交集 32 { 33 ans=ans+query(2*o,l,mid)/调用query函数递归 继续向子区间求值 34 } 35 if(qr>mid)/若右儿子与该要求的区间有交集 36 { 37 ans=ans+query(2*o+1,mid+1,r); 38 } 39 return ans;/返回左右儿子回馈的信息 40 } 41 42 int main() 43 { 44 scanf ql,qr; 45 int ans= query( 1 , 1 ,n); 46 } 47 48 49 /更新a[x]值,对应父节点的值也要更新; 50 51 void update(int o,int l,int r) 52 { 53 if(l==r)/更新叶子的sum值 54 { 55 sum[o] = v; 56 return; 57 } 58 int mid = (l+r) >> 1; 59 if(x <= mid)/如果x 在左儿子区域,往左儿子递归,更新左儿子对应的sum值。 60 { 61 update(2*o,l,mid); 62 } 63 else 64 { 65 update(2*o+1;mid+1,r); 66 } 67 sum[o]=sum[2*o]+sum[2*o+1]; 68 } 69 70 /离散化 节省内存 (比大小,只需知道相对大小就可以,排序后改变值为小,数组节省空间);
Description
Input
Output
Sample Input
Sample Output
1 #include<stdio.h> 2 #include<string.h> 3 4 int sum[150005],a[50005],ql,qr,id,add; 5 6 void build(int o,int l,int r) 7 { 8 if(l==r) 9 { 10 sum[o]=a[l]; 11 return; 12 } 13 int mid=(l+r)/2; 14 build(o*2,l,mid); 15 build(o*2+1,mid+1,r); 16 sum[o]=sum[o*2]+sum[o*2+1]; 17 } 18 19 int query(int o,int l,int r) 20 { 21 if(ql<=l && r<=qr) 22 return sum[o]; 23 int mid=(l+r)/2; 24 int ans=0; 25 if(ql<=mid){ 26 ans=ans+query(o*2,l,mid); 27 } 28 if(qr>mid){ 29 ans=ans+query(o*2+1,mid+1,r); 30 } 31 return ans; 32 } 33 34 void update(int o,int l,int r) 35 { 36 if(l==r){ 37 if(l==id) 38 sum[o]=sum[o]+add; 39 return; 40 } 41 int mid=(l+r)/2; 42 if(id<=mid) 43 { 44 update(2*o,l,mid); 45 } 46 else 47 { 48 update(2*o+1,mid+1,r); 49 } 50 sum[o]=sum[o*2]+sum[o*2+1]; 51 } 52 53 int main() 54 { 55 int i,j,k,T,N,ca=1; 56 bool flg; 57 char b[10]; 58 scanf("%d",&T); 59 while(T--) 60 { 61 memset(sum,0,sizeof(sum)); 62 flg=true; 63 scanf("%d",&N); 64 for(i=1;i<=N;i++) 65 { 66 scanf("%d",&a[i]); 67 } 68 build(1,1,N); 69 while(scanf("%s",&b)!=EOF) 70 { 71 72 if(b[0]==‘E‘) 73 break; 74 scanf("%d %d",&id,&add); 75 if(flg) 76 printf("Case %d:\n",ca),ca++,flg=false; 77 if(b[0]==‘A‘) 78 { 79 update(1,1,N); 80 } 81 else if(b[0]==‘S‘) 82 { 83 add=-add; 84 update(1,1,N); 85 } 86 else if(b[0]==‘Q‘) 87 { 88 ql=id,qr=add; 89 printf("%d\n",query(1,1,N)); 90 } 91 } 92 } 93 return 0; 94 }
原文:http://www.cnblogs.com/cyd308/p/4520873.html