xindoo is
always here

Light oj 1112 – Curious Robin Hood(树状数组)


题意:

   有n个数,有m组操作,1 i表示将第i个数先输出,然后置0, 2 i v 表示给第i个数加上v, 3 i j 表示求i 到 j 的和,注意,这里数组是从0开始的,而我们构造的树状数组是从1

开始的,使用在程序中要进行一定的处理。

//LA 1112 - Curious Robin Hood(树状数组)
//2013-04-13-08.22
#include <stdio.h>
#include <string.h>
const int maxn = 100010;

int a[maxn], b[maxn];
int n;

int lowbit(int x)
{
    return x&(-x);
}

void update(int x, int v)
{
    while (x <= n)
    {
        a[x] += v;
        x += lowbit(x);
    }
}

int getsum(int x)
{
    int sum = 0;
    while (x)
    {
        sum += a[x];
        x -= lowbit(x);
    }
    return sum;
}

int main()
{
    int t, m;
    scanf("%d",&t);
    for (int k = 1; k <= t; k++)
    {
        scanf("%d %d",&n, &m);
        memset(a, 0 ,sizeof(a));
        for (int i = 1; i <= n; i++)
        {
            scanf("%d",&b[i]);
            update(i, b[i]);
        }
        int op, i, v, j;
        printf("Case %d:\n",k);
        while (m--)
        {
            scanf("%d",&op);
            if (op == 1)
            {
                scanf("%d",&i);
                printf("%d\n",b[i+1]);
                update(i+1, -b[i+1]);
                b[i+1] = 0;
            }
            else if (op == 2)
            {
                scanf("%d %d",&i, &v);
                update(i+1, v);
                b[i+1] += v;
            }
            else
            {
                scanf("%d %d",&i,&j);
                printf("%d\n",getsum(j+1) - getsum(i));
            }
        }
    }
    return 0;
}


打赏
未经允许不得转载:XINDOO » Light oj 1112 – Curious Robin Hood(树状数组)
分享到: 更多 (0)

评论 抢沙发

xindoo

联系我联系我们

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

微信扫一扫打赏