【luogu P3368 树状数组2】 模板

题目链接：https://www.luogu.org/problemnew/show/P3368

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,m;
int tree[500070],a[500070];
int lowbit(int x)
{
    return x&-x;
}
void add(int k,int num)//给k位置的数值加num
{
    while(k<=n)
    {
        tree[k]+=num;
        k+=lowbit(k);
    }
}
int query(int x)
{
    int sum=0;
    for(int i=x;i>0;i-=i&(-i)) sum+=tree[i];
    return sum;
}

int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)  cin>>a[i];
    while(m--)
    {
        int k;
        cin>>k;
        if(k==1)
        {
            int x,y,z;
            cin>>x>>y>>z;
            add(x,z);           //维护查分数组
            add(y+1,-z);
        }
        else
        {
            int x;
            cin>>x;
            cout<<a[x]+query(x)<<endl;   //query()求的是改变的值，再加上原来的值就可以了
        }
    }
    return 0;
}