树状数组-实战一（洛谷P3368）

最新推荐文章于 2022-03-08 17:29:28 发布

原创最新推荐文章于 2022-03-08 17:29:28 发布 · 206 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#树状数组

本文详细介绍了如何使用树状数组进行单点更新及区间求和操作，并提供了完整的C++代码实现。通过实例演示了如何处理多个更新与查询操作，适用于解决区间更新与查询问题。

单点更新代码：

void add(int a , int b)
{
    while(a <= n)
    {
        tree[a] += b;
        a += lowbit(a);
    }
}

区间求和代码：

int sum(int x)
{
    int ans = 0;
    while(x != 0)
    {
        ans += tree[x];
        x -= lowbit(x);
    }
    return ans;
}

主代码：

#include<bits/stdc++.h>
using namespace std;
int n , m;
int tree[500001];
int lowbit(int k)
{
    return k & -k;
}
void add(int a , int b)
{
    while(a <= n)
    {
        tree[a] += b;
        a += lowbit(a);
    }
}
int sum(int x)
{
    int ans = 0;
    while(x != 0)
    {
        ans += tree[x];
        x -= lowbit(x);
    }
    return ans;
}
int main(void)
{
    int x ,y = 0 , z ,flag;
    int num;
    cin>>n>>m;
    for(int i = 1 ; i <= n ; i ++)
    {
        cin>>x;
        add(i,x-y);
        y = x;
    }
    for(int i = 1 ; i <= m ; i ++)
    {
        cin>>flag;
        if(flag == 1)
        {
            cin>>x>>y>>z;
            add(x,z);
            add(y+1,-z);
        }
        if(flag == 2)
        {
            cin>>x;
            cout<<sum(x)<<endl;
        }
    }
    return 0;
}

如有不懂参考如下：

https://blog.youkuaiyun.com/qq_33982232/article/details/81273634

https://blog.youkuaiyun.com/Small_Orange_glory/article/details/81290634