数据结构(树状数组)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;
}

 

posted @ 2013-07-25 11:22  执着追求的IT小小鸟  阅读(193)  评论(0编辑  收藏  举报