HDU - 1166 树状数组模板(线段树也写了一遍)

题意: 汉语题就不说题意了,用到单点修改和区间查询(树状数组和线段树都可以)
思路:
树状数组的单点查询,单点修改和区间查询。
树状数组是巧妙运用二进制的规律建树,建树就相当于单点修改。这里面用到一个lowbit的概念,将数的下标转化为二进制。找到值为1的最低位。前面的数舍去,判断这个数为几,那么这个数组的当前下标储存的值从这个下标开始往前的前几项和。
不是特别懂,先拿来用了
在树状数组里面求区间和,并不能直接求到,需要求两个区间之间的差。
树状数组:

#include
#include
int t,n,m,a[500010],c[500010];
int lowbit(int x)
{
    return x&(-x);
}
void update(int i,int k)//单点修改 加 树状数组初始化
{
    int j=i;
    while(j<=n)
    {
        c[j]+=k;
        j+=lowbit(j);
    }
}
int query(int l)//区间查询
{
    int i=l,sum=0;
    while(i>0)
    {
        sum+=c[i];
        i-=lowbit(i);
    }
    return sum;
}

int main()
{

    char s[100];
    int z=1;
    scanf("%d",&t);
    while(t--)
    {
        memset(a,0,sizeof(a));
        memset(c,0,sizeof(c));
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            update(i,a[i]);//建立树状数组
        }
//        printf("\n_______________\n");
//        for(int i=1;i<=n;i++)
//            printf("%d ",c[i]);
//        printf("\n_______________\n");
        int t1,t2,t3;
        printf("Case %d:\n",z++);
        while(1)
        {
            scanf("%s",s);
            if(s[0]=='E')
                break;
            if(s[0]=='Q')
            {
                scanf("%d%d",&t1,&t2);
                printf("%d\n",query(t2)-query(t1-1));
            }
            if(s[0]=='S')
            {
                scanf("%d%d",&t1,&t2);
                update(t1,t2*-1);
            }
            if(s[0]=='A')
            {
                scanf("%d%d",&t1,&t2);
                update(t1,t2);
            }
        }
    }
    return 0;
}

线段树:
#include
#include
int sum[300010];
void build(int l,int r,int o)
{
if(lr)
{
scanf("%d",&sum[o]);
return ;
}
int mid=(l+r)>>1;
build(l,mid,o<<1);
build(mid+1,r,o<<1|1);
sum[o]=sum[o<<1]+sum[o<<1|1];
}
int query(int x,int y,int l,int r,int o)
{
if(x<=l&&y>=r)
return sum[o];
int mid=(l+r)>>1;
int sum=0;
if(x<=mid) sum+=query(x,y,l,mid,o<<1);
if(y>mid) sum+=query(x,y,mid+1,r,o<<1|1);
return sum;
}
void update(int x,int y,int c,int l,int r,int o,int p)
{
if(x
l&&yr)
{
if(p
1)
sum[o]+=c;
else
sum[o]-=c;
return;
}
int mid=(l+r)>>1;
if(x<=mid) update(x,y,c,l,mid,o<<1,p);
if(y>mid) update(x,y,c,mid+1,r,o<<1|1,p);
sum[o]=sum[o<<1]+sum[o<<1|1];
}
int main()
{
int t,n,t1,t2,t3,z=1;
char a[101];
scanf("%d",&t);
while(t–)
{
printf(“Case %d:\n”,z++);
memset(sum,0,sizeof(sum));
scanf("%d",&n);
build(1,n,1);
while(1)
{
scanf("%s",a);
if(a[0]‘E’)
break;
if(a[0]
‘Q’)
{
scanf("%d%d",&t1,&t2);
printf("%d\n",query(t1,t2,1,n,1));
}
if(a[0]‘A’)
{
scanf("%d%d",&t1,&t2);
update(t1,t1,t2,1,n,1,1);
}
if(a[0]
‘S’)
{
scanf("%d%d",&t1,&t2);
update(t1,t1,t2,1,n,1,0);
}
}
}
return 0;
}


你可能感兴趣的:(数据结构)