樹狀數組(區間更新區間查詢)

#include<iostream>
#define ll long long
using namespace std;

const int maxn=200010;
int n,q,data;
long long  bit0[maxn],bit1[maxn];

void add(long long bit[],int i,int x){
    while (i<=n){
        bit[i]+=x;
        i+=i&-i; 
    } 
} 

ll sum(long long bit[],int i){
    ll ans=0;
    while (i>0){
        ans+=bit[i];
        i-=i&-i;
    }
    return ans;
}

int main(){
    cin>>n;
    for (int i=1;i<=n;i++){
        cin>>data;
        add(bit0,i,data);
    }   
    cin>>q;
    int tmp,x,y,v;
    while (q--){
        cin>>tmp;
        if (tmp==1){
            cin>>x>>y>>v;
            add(bit0,x,-v*(x-1));
            add(bit1,x,v);
            add(bit0,y+1,v*y);
            add(bit1,y+1,-v);
        }
        else{
            cin>>x>>y;
            ll ans=0;
            ans+=sum(bit1,y)*y+sum(bit0,y);
            ans-=sum(bit0,x-1)+sum(bit1,x-1)*(x-1);
            cout<<ans<<"\n";
        }
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章