AcWing 1264. 动态求连续区间和树状数组

最新推荐文章于 2024-09-13 14:59:21 发布

我也念过晚霞

最新推荐文章于 2024-09-13 14:59:21 发布

阅读量146

点赞数 1

分类专栏： AcWing 文章标签：算法

本文链接：https://blog.csdn.net/2303_79132118/article/details/137071885

版权

AcWing 专栏收录该内容

254 篇文章 0 订阅

订阅专栏

树状数组代码：

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

//第x个数加上v
int add(int x,int v)
{
    //因为树状数组的性质，加一个数，只影响logn个数，所有不用全加完
    //从当前位置开始加，每个间隔是lowbit(i)，一直加到最后
    for(int i=x;i<=n;i+=lowbit(i))
        tr[i]+=v;
}

//返回x的前缀和
int qurry(int x)
{
    //因为树状数组的性质，求前缀和，只用加logn个数，所有不用全加完
    //从当前位置开始累加，每个间隔是lowbit(i)，一直加到i==0停止
    int cnt=0;
    for(int i=x;i!=0;i-=lowbit(i))
        cnt+=tr[i];
    return cnt;
}

本题代码：

#include<bits/stdc++.h>

using namespace std;

const int N=100009;

int a[N],tr[N];

int n,m;

//每个数的间隔，背下来就行
int lowbit(int x)
{
    return x&-x;
}

//第x个数加上v
int add(int x,int v)
{
    //因为树状数组的性质，加一个数，只影响logn个数，所有不用全加完
    //从当前位置开始加，每个间隔是lowbit(i)，一直加到最后
    for(int i=x;i<=n;i+=lowbit(i))
        tr[i]+=v;
}

//返回x的前缀和
int qurry(int x)
{
    //因为树状数组的性质，求前缀和，只用加logn个数，所有不用全加完
    //从当前位置开始累加，每个间隔是lowbit(i)，一直加到i==0停止
    int cnt=0;
    for(int i=x;i!=0;i-=lowbit(i))
        cnt+=tr[i];
    return cnt;
}

int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=1;i<=n;i++)
        add(i,a[i]);//第i个数加上a[i]

    while(m--)
    {
        int k,x,y;
        scanf("%d%d%d",&k,&x,&y);
        if(k==0) printf("%d\n",qurry(y)-qurry(x-1));
        else add(x,y);
    }
    return 0;

}。