洛谷P3368树状数组2


基本思路,使用差分数组 arr[i]=a[i]-a[i-1];

#include<bits/stdc++.h>
#define maxn 1000000
using namespace std;
int arr[maxn],n;
int lowbit(int x)
{
    return x&(-x);
}
void add(int x,int d)
{
    while(x<=n)
    {
        arr[x]+=d;x+=lowbit(x);
    }
}
void range_add(int x,int y,int d)
{
    add(x,d);
    add(y+1,-d);
}
int sum(int x)
{
    int ans=0;
    while(x>0)
    {
        ans+=arr[x];x-=lowbit(x);
    }
    return ans;
}
int main()
{
    int m;cin>>n>>m;int now,pre=0;
    for(int i=1;i<=n;i++)
    {//差分
        cin>>now;
        add(i,now-pre);
        pre=now;
    }
    for(int i=1;i<=m;i++)
    {
        int op,x,y,k;
        cin>>op;
        if(op==1)
        {
            cin>>x>>y>>k;
            range_add(x,y,k);
        }
        else {
            cin>>x;
            cout<<sum(x)<<endl;
        }
    }
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/tldr/p/10897574.html