NYOJ 116 士兵杀敌 二

         http://acm.nyist.net/JudgeOnline/problem.php?pid=116

         树状数组已经看了好几天了,一直都是半懂半不懂,实在是忍无可忍了,今天晚上又看了看刘汝佳的白皮书,似乎明白了树状数组到底是怎么回事,果断写篇博客,明天要把线段树和字典树给搞定,不能再肉了~!

         树状数组可以很方便的查询任意区间内所有元素的和,还可以对树进行修改,时间复杂度位log(n)。

有两个很重的数组,A[n] 和 C[n],C 数组保存的是一点连续的和。

         对于i 节点,如果他是左孩子,那么父亲 节点的编号为 i + lowbit(i),如果是右孩子,那么父亲节点的编号为i - lowbit(i)。 这里的lowbit(i)指的是i 的最后一个1 对应的值。 如 10 = 1010.则lowbit(10) = 2。

         lowbit函数:

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

        有了C 数组,我们来求前缀Si 的和,妍在节点 i 往左走,边走边向上。把沿途的C[i]都加起来就是前缀和Si了。

        sum 函数:

int sum(int x)
{
    int num = 0;
    while(x > 0)
    {
        num += c[x];
        x -= lowbit(x);
    }
    return num;
}


        同样道理,当修改某个节点的值得时候,是沿着 节点 i 往右走,边走,边向上,把遇到的所有的C[i] 都给修改就行了。

        add 函数:

void add(int x, int y)
{
    while(x <= N)
    {
        c[x] += y;
        x += lowbit(x);
    }
}
        附上本题代码:

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
using namespace std;
int c[1000100],N;
int lowbit(int x)
{
    return x & (-x);
}
int sum(int x)
{
    int num = 0;
    while(x > 0)
    {
        num += c[x];
        x -= lowbit(x);
    }
    return num;
}
void add(int x, int y)
{
    while(x <= N)
    {
        c[x] += y;
        x += lowbit(x);
    }
}
int main()
{
    int M,i,j,a,b;
    char s[6];
    memset(c,0,sizeof(c));
    scanf("%d%d",&N,&M);
    for(i = 1; i <= N; i++)
    {
        scanf("%d",&a);
        add(i,a);
    }
    for(i = 0; i < M; i++)
    {
        scanf("%s%d%d",s,&a,&b);
        if(s[0] == 'Q') printf("%d\n",sum(b) - sum(a-1));
        else add(a,b);
    }
    return 0;
}



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值