线段树初探

#include<iostream>
#include <cstdio>
#include <cstring>
using namespace std;
typedef long long ll;
const int maxn = 1e5+100;

struct node{
    int l,r;//区间端点 
    int len;//区间长度 
    ll sum;//区间和 
    ll lazy;//懒标记 
}tree[maxn<<2];

ll a[maxn]; //原数组 

void pushUp(int rt){
    tree[rt].sum = tree[rt<<1].sum + tree[rt<<1|1].sum;
}

void build(int rt, int l, int r){
    tree[rt].l = l;//左端点赋值 
    tree[rt].r = r;//右端点赋值  
    tree[rt].lazy = 0;//懒标记初始值记为0 
    
    tree[rt].len = r - l + 1;//区间长度赋值 
    
    if(l == r){//到了叶子节点 
        //tree[rt].sum = 1;//初始化为1 
        tree[rt].sum = a[l] ;//原来的数组a[] 
        return; 
    }
    
    int mid = (l + r) >> 1;
    build(rt<<1 , l, mid);//递归建立左子树 
    build(rt<<1|1, mid+1, r);//递归建立右子树 
    
    pushUp(rt);//回溯更新所有区间和 
}

void pushDown(int rt){
    if(tree[rt].lazy){
        
        //更新子节点的sum 
        tree[rt<<1].sum += tree[rt].lazy * tree[rt<<1].len;
        tree[rt<<1|1].sum += tree[rt].lazy * tree[rt<<1|1].len;
        //下推标记 
        tree[rt<<1].lazy += tree[rt].lazy;
        tree[rt<<1|1].lazy += tree[rt].lazy;
        //根节点懒标记清零 
        tree[rt].lazy = 0;
    }
}


void update(int rt, int l , int r, int val){
    
    //如果待更新区间[l,r]恰好是当前结点的区间则直接更新当前结点 
    if(tree[rt].l == l && tree[rt].r == r){
    
        tree[rt].sum += val * tree[rt].len;
        tree[rt].lazy += val;
        return ;
    }
    //告诉我的左右孩子要更新了，爸爸已经更新了 
    pushDown(rt);
    
    int mid = (tree[rt].l + tree[rt].r) >> 1;
    
    // 递归更新 ，给待更新区间 [l,r]分段更新直到满足上面的可以直接更新的条件
    if(l > mid){
        update(rt<<1|1, l , r, val);
    }
    else if(r <= mid){
        update(rt<<1, l , r, val);
    }
    else{
        update(rt<<1, l , mid, val);
        update(rt<<1|1, mid+1 , r, val);
    }
    //向上保持正确 
    pushUp(rt);
    
}

ll query(int rt, int l , int r){
    ll ans = 0;
    
    if(tree[rt].l ==l && tree[rt].r == r){ //如果待查区间[l,r]恰好是当前结点的区间则直接返回 
        return tree[rt].sum;
    }
    pushDown(rt);//下推懒标记告诉我的左右孩子要更新了
     
    //递归查询 ，给待查区间 [l,r]分段查询直到满足上面的可以直接返回的条件 
    int mid = (tree[rt].l + tree[rt].r) >> 1;
    if(l > mid){ 
        ans += query(rt<<1|1, l , r);
    }
    else if(r <= mid){
        ans += query(rt<<1, l , r);
    }
    else{
        ans += query(rt<<1, l , mid) + query(rt<<1|1, mid+1 , r); 
    }
    
    return ans;
}

int main()
{
    int n,m;
    scanf ("%d %d",&n,&m);
    
    for(int i=1; i<=n; i++){
        scanf ("%lld",&a[i]);
    } 
    build(1,1,n);
    int x,y,z;
    char op[3];
    while(m--)
    {
        scanf ("%s",op);
        if (op[0] == ‘C‘)
        {
            scanf ("%d %d %d",&x,&y,&z);
            update(1,x,y,z);
        }
        else
        {
            scanf ("%d %d",&x,&y);
            printf ("%lld\n",query(1,x,y));
        }
    }
    return 0;
}