BIT-3

题目描述

给定数组�1,�2...��a​1​​,a​2​​...a​n​​，进行q次操作，操作有两种：

1 l r k：将��a​i​​ ~ ��a​r​​ 每个数都加上 k；

2 l r：求��+...+��a​l​​+...+a​r​​

输入描述

第一行：输入两个数 n，q，表示给定数组的长度和操作数

第二行：输入n个数，表示长度为n的给定数组

接下来q行：每行输入表示如题的某一种操作

输出描述

对于每个2操作，输出对应结果

输入样例

1. 5 3
2. 2 6 6 1 1
3. 2 1 4
4. 1 2 5 10
5. 2 1 3

输出样例

1. 15
2. 34

数据描述

1≤�,�≤1061≤n,q≤10​6​​

∣��∣≤106∣a​i​​∣≤10​6​​

对于输入：保证1≤�≤�≤�,∣�∣≤1061≤l≤r≤n,∣k∣≤10​6​​

代码

#include<iostream>
using namespace std;
long long n,x,c1[1000005],c2[1000005],q,L,R,k;
int opt;
const int N=1e6+5;
#define lb(x) (x&(-x))
void update(int x,long long  y){
    for(int i=x;i<=n;i+=lb(i)) c1[i]+=y,c2[i]+=y*x;
}
long long ask(int x){
    long long ans=0;
    for(int i=x;i;i-=lb(i)) ans+=c1[i];
    ans*=(x+1);
    for(int i=x;i;i-=lb(i)) ans-=c2[i];
    return ans;
}
int main(){
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    cin>>n>>q;
    for(int i=1;i<=n;i++) {
        cin>>x;
        update(i,x),update(i+1,-x);
    }
    while(q--){
        cin>>opt;
        if(opt==1){
            cin>>L>>R>>k;
            update(L,k),update(R+1,-k);
        }
        else{
            cin>>L>>R;
            cout<<ask(R)-ask(L-1)<<"\n";
        }
    }
    return 0;
}