树状数组(拓展应用)

给定长度为 N 的数列 A，然后输入 M 行操作指令。

第一类指令形如 C l r d，表示把数列中第 l∼r 个数都加 d。

第二类指令形如 Q x，表示询问数列中第 x 个数的值。

对于每个询问，输出一个整数表示答案。

输入格式

第一行包含两个整数 N 和 M。

第二行包含 N个整数 A[i]

接下来 M 行表示 M 条指令，每条指令的格式如题目描述所示。

输出格式

对于每个询问，输出一个整数表示答案。

每个答案占一行。

数据范围

1≤N,M≤105
|d|≤10000|
|A[i]|≤109|

输入样例：

10 5
1 2 3 4 5 6 7 8 9 10
Q 4
Q 1
Q 2
C 1 6 3
Q 2

输出样例：

4
1
2
5

前缀和差分都能够将对区间操作为0(n)的时间变为0(1),但是当需要对单点进行查询等操作时,反倒捉襟见肘,那么树状数组就是将操作变为了0(logn),其实就可以想像为将当前数组copy到另一个数组,这个数组就叫做树状数组,那么对于本题,当要对某一区间进行操作时就需要我们把原数组的差分数组存入到树状数组,之后的任何操作就都可以完成了

#include <iostream>

using namespace std;
typedef long long ll;
constexpr int N=1e5+7;
int n,m;
ll lowbit(ll x){
    return x&-x;
}
ll a[N],tr[N];
void add(ll x,ll c){
    for(ll i=x;i<=n;i+= lowbit(i)) tr[i]+=c;
}
ll sum(ll x){
    ll res=0;
    for(ll i=x;i;i-= lowbit(i)) res+=tr[i];
    return res;
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
    for(int i=1;i<=n;i++){
        add(i,a[i]-a[i-1]);
    }
    while(m--){
        char op[2];
        int l,r,d;
        scanf("%s%d",op,&l);
        if(*op=='C'){
            scanf("%d%d",&r,&d);
            add(l,d), add(r+1,-d);
        }
        else{
            printf("%lld\n", sum(l));
        }
    }
}

如果改为

1. C l r d，表示把 A[l],A[l+1],…,A[r]都加上 d。
2. Q l r，表示询问数列中第 l∼r个数的和。 

 

 

#include <iostream>

using namespace std;
typedef long long ll;
constexpr int N=1e5+7;
int n,m;
ll lowbit(ll x){
    return x&-x;
}
ll a[N],tr1[N],tr2[N];
void add(ll tr[],ll x,ll c){
    for(ll i=x;i<=n;i+= lowbit(i)) tr[i]+=c;
}
ll sum(ll tr[],ll x){
    ll res=0;
    for(ll i=x;i;i-= lowbit(i)) res+=tr[i];
    return res;
}
ll qnum(ll x){
    return sum(tr1,x )*(x+1)- sum(tr2,x);
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
    for(int i=1;i<=n;i++){
        ll b=a[i]-a[i-1];
        add( tr1,i,b);
        add( tr2,i,(ll)b*(ll)i);
    }
    while(m--){
        char op[2];
        ll l,r,d;
        scanf("%s%lld%lld",op,&l,&r);
        if(*op=='C'){
            scanf("%lld",&d);
            add(tr1,l,d), add(tr1,r+1,-d);
            add(tr2,l,l*d), add(tr2,r+1,-d*(r+1));
        }
        else{
            printf("%lld\n", qnum(r)- qnum(l-1));
        }
    }
}