洛谷P3368 树状数组(区间修改,单点查询)

题目传送门

今天刚学树状数组,操作还不是很熟练。

对于区间修改,如果我们还是像单点修改一样,要把区间内所有都操作一遍是不现实的。
所以在只需要单点查询的时候,树状数组维护的不是原数组,而是原数组的差分。
差分的知识在此便不再赘述。

#include<bits/stdc++.h>
using namespace std;
const int N=5e5+5;
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
const double EPS=1e-6;
typedef long long ll;
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
//#define int long long
int a[N],d[N],c[N];
int n,m;
int lowbit(int x)
{
    return x&-x;
}
void add(int x,int k)
{
    for(int i=x;i<=n;i+=lowbit(i))
    {
        c[i]+=k;
    }
}
int sum(int x)
{
    int s=0;
    for(int i=x;i;i-=lowbit(i))
    {
        s+=c[i];
    }
    return s;
}
signed main()
{
    IOS;
    //freopen("","r",stdin);
    //freopen("","w",stdout);
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        d[i]=a[i]-a[i-1];
        add(i,d[i]);
    }
    while(m--)
    {
        int t;
        cin>>t;
        if(t==1)
        {
            int x,y,k;
            cin>>x>>y>>k;
            add(x,k);
            add(y+1,-k);
        }
        else
        {
            int x;
            cin>>x;
            int t=sum(x);
            cout<<t<<endl;
        }
    }
}

发布了93 篇原创文章 · 获赞 9 · 访问量 4205

猜你喜欢

转载自blog.csdn.net/Joker_He/article/details/104270538