洛谷 P3368 【模板】树状数组 2

最新推荐文章于 2024-06-05 21:28:18 发布

wineandchord

最新推荐文章于 2024-06-05 21:28:18 发布

阅读量81

点赞数

分类专栏：线段树/树状数组模板

本文链接：https://blog.csdn.net/m0_37795244/article/details/105309308

版权

模板同时被 2 个专栏收录

13 篇文章 0 订阅

订阅专栏

线段树/树状数组

7 篇文章 0 订阅

订阅专栏

本文介绍了一种处理数列区间修改与单点查询的高效算法。通过使用差分数组和树状数组（Binary Indexed Tree），可以快速更新指定区间的数值，并查询任意位置的最新值。文章提供了完整的C++实现代码，包括lowbit、add和query函数，适用于解决复杂的数据结构问题。

摘要由CSDN通过智能技术生成

给一个数列，进行以下操作：

某区间每个数 $+ x$
求某位置的值

区间修改，单点查询。用数组 b 来存放修改产生的差分。

代码如下：

#include<iostream>
#include<cstdio>
#define MAXN 500010
using namespace std;
typedef long long ll;
int n,m,t,x,y;
ll a[MAXN],b[MAXN],k;
int lowbit(int x){return x&-x;}
void add(int x,ll k){
    while(x<=n){
        b[x]+=k;
        x+=lowbit(x);
    }
}
ll query(int x){
    ll res=0;
    while(x>0){
        res+=b[x];
        x-=lowbit(x);
    }
    return res;
}
int main(){
#ifdef WINE
    freopen("data.in","r",stdin);
#endif
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%lld",&a[i]);
    while(m--){
        scanf("%d%d",&t,&x);
        if(t==1){
            scanf("%d%lld",&y,&k);
            add(x,k);add(y+1,-k);
        }else printf("%lld\n",a[x]+query(x));
    }
    return 0;
}