数据结构(树状数组)HDU1166
http://acm.hdu.edu.cn/showproblem.php?pid=1166
树状数组的运用,关键在于理解树状数组的建立及相关操作
操作主要分四个:
一:管辖域的求解
就是2的k次方,有公式:cn=a(n-a^k+1)+.........+an(其中 k 为 n 的二进制表示中从右往左数的 0 的个数)。
则可以用函数lowbit(n)直接求出
二,单个数的调整
当有一个数发生改变时,管辖到它的数也要做相关的调整
三,求和
将管辖到的数加上未管辖到的数
四,建树
有公式:
c[i]=sum[i]-sum[i-lowbit(i)];i从1开始
#include<stdio.h> #include<string.h> int a[50010],c[50010]; int lowbit(int x) { return x&(-x);// } int sum(int x) { int sum=0; while(x>0) { sum+=c[x]; x-=lowbit(x);//找寻它未管辖到的数 } return sum; } void change(int i,int x,char *s,int n) { while(i<=n) { c[i]+=strcmp(s,"Add")==0?x:(-x); i+=lowbit(i);//找寻管辖到它的数 } } int main() { int t,i,x,n,si,sj,j=1; char key[20]; scanf("%d",&t); while(t--) { scanf("%d",&n); a[0]=c[0]=0; for(i=1;i<=n;i++) { scanf("%d",&a[i]); a[i]+=a[i-1]; c[i]=a[i]-a[i-lowbit(i)];//建树 } printf("Case %d:\n",j++); while(scanf("%s",key),strcmp(key,"End")) { scanf("%d%d",&si,&sj); if(strcmp(key,"Add")==0||strcmp(key,"Sub")==0) { change(si,sj,key,n); } else { x=sum(sj)-sum(si-1); printf("%d\n",x); } } } return 0; }