P3368 【模板】树状数组 2

P3368 【模板】树状数组 2

题目传送门

P3368 【模板】树状数组 2

题目大意

给你一个长度为n的数组
操作 1： 格式：1 x y k 含义：将区间 $[x,y]$ 内每个数加上 $k$；
操作 2： 格式：2 x 含义：输出第 $x$ 个数的值。

思路

区间更新，单点查询的树状数组

AC Code

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
using namespace std;
#define endl '\n'
#define INF 0x3f3f3f3f
#define int long long
// #define TDS_ACM_LOCAL
const int N=5e5 +9;
int n, m;
int a[N],bit[N]; //对应原数组和树状数组

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

void updata(int i,int k){    //在i位置加上k
    while(i <= n){
        bit[i] += k;
        i += lowbit(i);
    }
}

int getsum(int i){        //求A[1 - i]的和
    int res = 0;
    while(i > 0){
        res += bit[i];
        i -= lowbit(i);
    }
    return res;
}

//差分建树，这样更新区间时变为了a[x]和a[y+1]的更新
void solve(){
    cin>>n>>m;
    for(int i = 1; i <= n; i++){
        cin>>a[i];
        updata(i,a[i] - a[i-1]);   //输入初值的时候，也相当于更新了值
    }
    while(m--){
        int flag, x, y, k;
        cin>>flag;
        if(flag==1){
            cin>>x>>y>>k;
            //[x,y]区间内加上k
            updata(x,k);    //A[x] - A[x-1]增加k
            updata(y+1,-k);        //A[y+1] - A[y]减少k
        }
        else if(flag==2){
            cin>>x;
            cout<<getsum(x)<<endl;
        }
    }
    return ;
}

signed main(){
	ios::sync_with_stdio(0);
	cin.tie(0), cout.tie(0);
#ifdef TDS_ACM_LOCAL
    freopen("D:\\VS code\\.vscode\\testall\\in.txt", "r", stdin);
    freopen("D:\\VS code\\.vscode\\testall\\out.txt", "w", stdout);
#endif
    solve();
    return 0;
}