您的位置:首页 > 其它

HDU 1166 树状数组

2015-08-07 20:43 337 查看
以前用线段树写的,编码复杂,效率也低。今天刚学习了树状数组,发现这道题可以用树状数组来写,而且时间也节省了一半,编码复杂度和空间复杂度就更不用说了。


#include<stdio.h>
#include<string.h>
#define lowbit(x) x&(-x)
int a[50010],n;
void add(int x,int d)
{
    while(x<=n)
    {
        a[x]+=d;
        x+=lowbit(x);
    }
}
int q(int x)
{
    int sum=0;
    for(int i=x;i;i-=lowbit(i))
        sum+=a[i];
    return sum;
}
int main()
{
    int t,i,tmp,l,r,cas=0;
    char s[100];
    scanf("%d",&t);
    while(t--)
    {
        printf("Case %d:\n",++cas);
        memset(a,0,sizeof(a));
        scanf("%d",&n);
        for(i=1;i<=n;i++)
        {
            scanf("%d",&tmp);
            add(i,tmp);
        }
        getchar();
        while(scanf("%s",s)!=EOF&&s[0]!='E')
        {
            scanf("%d%d",&l,&r);
            if(s[0]=='A')
                add(l,r);
            else if(s[0]=='S')
                add(l,-r);
            else
                printf("%d\n",q(r)-q(l-1));
        }
    }
    return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: