[算法]树状数组 2(差分)

P3368 【模板】树状数组 2

#include<iostream>
#include<cstdio>
#include<algorithm>
#define ll long long
using namespace std;
const int maxn = 5e5 + 10;

ll cnt[maxn];
ll a[maxn];
ll n, m;
ll lowbit(ll x) {
    // x 的二进制表示中，最低位的 1 的位置。
// lowbit(0b10110000) == 0b00010000
//          ~~~^~~~~
// lowbit(0b11100100) == 0b00000100
//   
    return x & -x;
}
//差分思想
inline void add(ll x, ll k)
{
    while (x <= n)
    {
        cnt[x] += k;
        x += lowbit(x);
    }
}
ll getsum(ll x)
{
    ll ans = 0;
    while (x > 0)
    {
        ans += cnt[x];
        x -= lowbit(x);
    }
    return ans;
}
int main()
{
    scanf("%lld%lld", &n, &m);
    for (ll i = 1; i <= n; i++)
    {
        scanf("%lld", &a[i]);
    }
    while (m--)
    {
        ll op, x, y, k;
        cin >> op;
        if (op == 1)
        {
            scanf("%lld%lld%lld", &x, &y, &k);
            add(x, k); add(y + 1, -k);//如果区间修改2到4位之后求第3位，我们就可以加上修改的数。而如果求第5位，
            //我们只需要在第3位的基础上减去修改的数就可以了。举个例子：在数组第2位和第4位之间加上2，
            //只需要将数组从0 0 0 0 0变成0 2 0 0 -2即可。当询问第3位时，答案就是输入的3的值+0+2+0即可。当询问5时，
            //答案就是输入的5的值+0+2+0+0+-2=输入的5的值+0。这就是解法了！
        }
        else
        {
            scanf("%lld", &x);
            printf("%lld\n", a[x] + getsum(x));
        }
    }
    return 0;
}