hdu 1166 敌兵布阵（用树状数组）

敌兵布阵

      仍然是敌兵布阵那题，题目大意：给你一串数，然后会根据题意选择一点增加或减少，或者询问某区间的人数有多少？之前用线段树写了，而这题可以用树状树状来做，更加方便更加快速。

说下树状数组的三个主要函数：（神一样的函数，不只这些用处！！！）

1.lowbit(int i)

2.update(int i, int x)

3.sum(int i)

链接：http://acm.hdu.edu.cn/showproblem.php?pid=1166

代码：

#include <iostream>
#include <stdio.h>
#include <memory.h>
using namespace std;

int n, a[50005];
char sh[15];

int lowbit(int i)   //树状数组最巧妙之处：i&(-i)
{
    return i&(-i);
}

void update(int i, int val) //更新函数
{
    while(i <= n)
    {
        a[i] += val;
        i += lowbit(i);
    }
}

int sum(int i)      //求和函数
{
    int sum = 0;
    while(i > 0)
    {
        sum += a[i];
        i -= lowbit(i);
    }
    return sum;
}

int main()
{
    int i, val, t, x, y, zz = 1;
    scanf("%d", &t);
    while(t--)
    {
        memset(a, 0, sizeof(a));
        scanf("%d", &n);
        for(i = 1; i <= n; i++)
        {
            scanf("%d", &val);
            update(i, val);
        }
        printf("Case %d:\n", zz++);
        while(scanf("%s", sh))
        {
            if(sh[0] == 'E') break;
            scanf("%d %d", &x, &y);
            if(sh[0] == 'A') update(x, y);
            else if(sh[0] == 'S') update(x, -y);
            else printf("%d\n", sum(y)-sum(x-1));   //两段区间和相减
        }
    }

    return 0;
}

 

评论

3 楼 panyanyany 2011-08-07  

神啊，大神啊！膜拜大神啊！

2 楼 sgeteternal 2011-08-07  

爽+1

1 楼 基德KID.1412 2011-08-07  

爽