AcWing 243. 一个简单的整数问题2 (树状数组,区间更新/询问)

AcWing 243. 一个简单的整数问题2  (树状数组,区间更新/询问)

  • 题意:区间更新,区间询问.

  • 题解;对于区间更新,我们还是用差分数组\(b_i\)来更新,区间询问时,我们的答案是:\(\sum_{i=l}^{r}\sum_{j=1}^{i}b_j\),

    AcWing 243. 一个简单的整数问题2  (树状数组,区间更新/询问)

    所以,我们搞两个树状数组维护\(b_i\)和\(i*b_i\)即可.

  • 代码:

    #define int long long 
    
    int n,m;
    int a[N];
    int c1[N],c2[N];
    
    int lowbit(int x){
        return x&(-x);
    }
    
    void updata1(int i,int k){
        while(i<=n){
            c1[i]+=k;
            i+=lowbit(i);
        }
    }
    
    void updata2(int i,int k){
        while(i<=n){
            c2[i]+=k;
            i+=lowbit(i);
        }
    }
    
    int get_sum1(int i){
        int res=0;
        while(i){
            res+=c1[i];
            i-=lowbit(i);
        }
        return res;
    }
    
    int get_sum2(int i){
        int res=0;
        while(i){
            res+=c2[i];
            i-=lowbit(i);
        }
        return res;
    }
    
    signed main() {
        ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
        cin>>n>>m;
    
        rep(i,1,n){
            cin>>a[i];
            updata1(i,a[i]-a[i-1]);
            updata2(i,i*(a[i]-a[i-1]));
        }
    
        rep(i,1,m){
            char op;
            cin>>op;
            if(op=='Q'){
                int l,r;
                cin>>l>>r;
                int cur1=get_sum1(r)*(r+1)-get_sum2(r);
                int cur2=get_sum1(l-1)*l-get_sum2(l-1);
                cout<<cur1-cur2<<'\n';
            }
            else{
                int l,r,d;
                cin>>l>>r>>d;
                updata1(l,d);
                updata2(l,l*d);
                updata1(r+1,-d);
                updata2(r+1,(r+1)*-d);
            }
        }
    
     
        return 0;
    }
    
上一篇:9102年:手写一个Vue的脚手架


下一篇:qt creator 使用Ui文件的问题