树状数组/线段树 243. 一个简单的整数问题2

243. 一个简单的整数问题2

给定一个长度为 N 的数列 A，以及 M 条指令，每条指令可能是以下两种之一：

C l r d，表示把 A[l],A[l+1],…,A[r] 都加上 d。
Q l r，表示询问数列中第 l∼r 个数的和。
对于每个询问，输出一个整数表示答案。

输入格式
第一行两个整数 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 4
Q 1 10
Q 2 4
C 3 6 3
Q 2 4
输出样例：
4
55
9
15

本题使用了差分与前缀和的性质，关系如上图，开两个数组分别维护 b[i] 与 i*b[i]来进行区间查询与修改。树状数组写法

#include<cstdio>
using namespace std;
#define lowbit(x)x&-x
typedef long long ll;
const int N = 1e5+10;
ll tr1[N],tr2[N],n,m,a[N];
inline void update(ll tr[],int u,ll x){for(;u<=n;u+=lowbit(u))tr[u]+=x;}
inline ll sum(ll tr[],int u){ll res(0);for(;u;u-=lowbit(u))res+=tr[u];return res;}
inline ll get(int u){return sum(tr1,u)*(u+1)-sum(tr2,u);}
int main(){
    scanf("%d%d",&n,&m);
    for(int i(1);i<=n;i++){
        scanf("%d",&a[i]);
    }
    for(int i(1);i<=n;i++){
        int b=a[i]-a[i-1];
        update(tr1,i,b);
        update(tr2,i,(ll)b*i);
    }
    while(m--){
        int l,r,d;
        char op[2];
        scanf("%s %d %d",op,&l,&r);
        if(*op=='Q'){
         printf("%lld\n",get(r)-get(l-1));
        }
        else {
            scanf("%d",&d);
            update(tr1,l,d),update(tr1,r+1,-d);
            update(tr2,l,(ll)d*l),update(tr2,r+1,(ll)(r+1)*-d);
        }
    }
}

线段树写法，加入add标记

#include<cstdio>
#pragma optimize(3)
using namespace std;
typedef long long ll;
inline ll lowbit(ll x){return x&-x;}
const int N = 1e5+10;
struct xds{
    int l,r;
    ll sum,add;
}tr[N*4];
int m,n,a[N];
inline void pushdown(int u){
      auto &root=tr[u],&l=tr[u<<1],&r=tr[u<<1|1];
      if(root.add){
          l.add+=root.add,l.sum+=(ll)(l.r-l.l+1)*root.add;
          r.add+=root.add,r.sum+=(ll)(r.r-r.l+1)*root.add;
           root.add=0;
      }
}
inline void pushup(xds&u,xds&l,xds&r){
    u.sum=l.sum+r.sum;
}
inline void pushup(int u){
    pushup(tr[u],tr[u<<1],tr[u<<1|1]);
}
inline void build(int u,int l,int r){
    if(l==r){
        tr[u]={l,r,a[r],0};
    }
    else {
     tr[u]={l,r};
     int mid=l+r>>1;
     build(u<<1,l,mid),build(u<<1|1,mid+1,r);
         pushup(u);   
    }
}
inline ll query(int u,int l,int r){
    if(tr[u].l>=l&&tr[u].r<=r)return tr[u].sum;
    else {
        pushdown(u);
        int mid=tr[u].l+tr[u].r>>1;
        ll sum=0;
        if(l<=mid)sum+=query(u<<1,l,r);
        if(r>mid)sum+=query(u<<1|1,l,r);
        return sum;
    }
}
inline void modify(int u,int l,int r,int d){
    if(tr[u].l>=l&&tr[u].r<=r){
        tr[u].sum+=(ll)(tr[u].r-tr[u].l+1)*d;
        tr[u].add+=d;
    }
    else {
        int mid=tr[u].l+tr[u].r>>1;
        pushdown(u);
        if(l<=mid)modify(u<<1,l,r,d);
        if(r>mid)modify(u<<1|1,l,r,d);
        pushup(u);
    }
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i(1);i<=n;i++)scanf("%d",&a[i]);
    build(1,1,n);
    while(m--){
        char op[2];
        int l,r,x;
        scanf("%s %d %d",op,&l,&r);
        if(*op=='Q'){
          printf("%lld\n",query(1,l,r));
        }
        else {
            scanf("%d",&x);
            modify(1,l,r,x);
        }
    }
}