线段树——简单的整数问题2

简单的整数问题2

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

1、“C l r d”，表示把 A[l],A[l+1],…,A[r] 都加上 d。

2、“Q l r”，表示询问 数列中第 l~r 个数的和。

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

输入格式
第一行两个整数N,M。

第二行N个整数A[i]。

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

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

每个答案占一行。

数据范围
$1\le N,M\le 10^5$,
$|d|\le 10000$,
$|A[i]|\le 1000000000$
输入样例：
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
多写写，找手感，固定模板。

#include <bits/stdc++.h>
//#define int long long
const int N=1e5+7;
#define lson u<<1
#define rson u<<1|1
using namespace std;
struct Node{
    int l,r;
    long long sum,add;
}tr[N<<2];
int n,m,a[N];
void pushup(Node &u,Node &l,Node &r){
    u.sum=l.sum+r.sum;
}
void pushup(int u)
{
    pushup(tr[u],tr[lson],tr[rson]);
}
void build(int u,int l,int r)
{
//    cout<<u<<' '<<l<<' '<<r<<endl;
    if(l==r){
        tr[u]={l,r,a[l],0};
        return ;
    }
    int mid=l+r>>1;
    tr[u]={l,r};
    build(lson,l,mid);
    build(rson,mid+1,r);
    pushup(u);
}
void pushdown(int u)
{
    if(tr[u].add){
        tr[lson].add+=tr[u].add;
        tr[rson].add+=tr[u].add;
        tr[lson].sum+=(tr[lson].r-tr[lson].l+1)*tr[u].add;
        tr[rson].sum+=(tr[rson].r-tr[rson].l+1)*tr[u].add;
        tr[u].add=0;
    }
}
Node query(int u,int l,int r)
{
    if(tr[u].l>=l&&tr[u].r<=r){
        return tr[u];
    }
    pushdown(u);
    int mid=tr[u].l+tr[u].r>>1;
    if(r<=mid) return query(lson,l,r);
    else if(l>mid) return query(rson,l,r);
    else{
        auto left=query(lson,l,r);
        auto right=query(rson,l,r);
        Node res;
        pushup(res,left,right);
        return res;
    }
}
void modify(int u,int l,int r,int c)
{
    if(tr[u].l>=l&&tr[u].r<=r){
        tr[u].add+=c;
        tr[u].sum+=(tr[u].r-tr[u].l+1)*c;
        return;
    }
    int mid=tr[u].l+tr[u].r>>1;
    pushdown(u);//造成一个区间add信息不同所以必须先分裂
    if(l<=mid) modify(lson,l,r,c);
    if(r>mid) modify(rson,l,r,c);
    pushup(u);
}
int main(){
//    cout<<"--"<<endl;
    scanf("%lld%lld",&n,&m);
    for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
    build(1,1,n);
//    cout<<"--"<<endl;
    for(int i=1;i<=m;i++){
        char op[3];
        scanf("%s",op);
        if(*op=='Q'){
            int l,r; scanf("%lld%lld",&l,&r);
            printf("%lld\n",query(1,l,r).sum);
        }else {
            int l,r,c; scanf("%lld%lld%lld",&l,&r,&c);
            modify(1,l,r,c);
        }
    }
    return 0;
}