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

发布时间 2023-03-27 21:13:31作者: 高尔赛凡尔娟

https://www.luogu.com.cn/problem/P3368

题目大意:

如题,已知一个数列,你需要进行下面两种操作:

1.将某区间每一个数加上x;

2.求出某一个数的值。
输入 #1复制
5 5
1 5 4 2 3
1 2 4 2
2 3
1 1 5 -1
1 3 5 7
2 4
输出 #1复制
6
10

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef pair<LL,LL> PII;
const LL MAXN=1e18,MINN=-MAXN,INF=0x3f3f3f3f;
const LL N=2e6+10,M=2023;
const LL mod=998244353;
const double PI=3.1415926535;
#define endl '\n'
int n,m;
int a[N],b[N],tr[N];
int lowbit(int x)
{
    return x&(-x);
}
void add(int x,int v)
{
    for(int i=x;i<=n;i+=lowbit(i))
    {
        tr[i]+=v;
    }
}
int query(int x)
{
    int res=0;
    for(int i=x;i;i-=lowbit(i))
    {
        res+=tr[i];
    }
    return res;
}
int main()
{
    cin.tie(0); cout.tie(0); ios::sync_with_stdio(false);
    int T=1;
    //cin>>T;
    while(T--)
    {
        cin>>n>>m;
        for(int i=1;i<=n;i++)
        {
            cin>>a[i];
        }
        for(int i=1;i<=n;i++)
        {
            b[i]=a[i]-a[i-1];
            add(i,b[i]);
        }
        while(m--)
        {
            LL op;
            cin>>op;
            if(op==1)
            {
                LL x,y,k;
                cin>>x>>y>>k;
                add(x,k);
                add(y+1,-k);
            }
            else if(op==2)
            {
                LL x;
                cin>>x;
                cout<<query(x)<<endl;
            }
        }
    }
    return 0;
}