【模板】树状数组2

tech2024-04-21  9

前往:我自己搭建的博客

题目

洛谷P3368树状数组2

题解

将普通树状数组操作和差分思想结合,就可以实现区间修改和单点查询。若对区间[l,r]进行修改,则在l处修改,并在r+1处反向修改。在查询时,询问x处的值,则求[1,x]的和,若x在[l,r]内,则会累加l处的修改,若x在r右侧,则r+1处的反向修改会抵消l处的修改。

代码

#include <bits/stdc++.h> using namespace std; typedef long long ll; const int maxn=5e5+5; int n,m; int sum[maxn],a[maxn]; //a[]存初始数据 inline int lowbit(int x) {return x&(-x);} inline void add(int x,int y) {for( ;x<=n;x+=lowbit(x)) sum[x]+=y;} inline int ask(int x) {int ans=0; for( ;x;x-=lowbit(x)) ans+=sum[x]; return ans;} int main() { scanf("%d%d",&n,&m); for(int i=1;i<=n;i++) scanf("%d",&a[i]); for(int i=1;i<=m;i++) { int t; scanf("%d",&t); if(t==1) { int x,y,k; scanf("%d%d%d",&x,&y,&k); add(x,k); add(y+1,-k); } else { int x; scanf("%d",&x); printf("%d\n",a[x]+ask(x)); //注意加上原始数据 } } return 0; }
最新回复(0)