训练之线段树A Simple Problem with Integers

A Simple Problem with Integers Crawling in process... Crawling failed Time Limit:5000MS     Memory Limit:131072KB     64bit IO Format:%lld & %llu

Submit Status

Description

Input

Output

Sample Input

Sample Output

Hint

Description

给出了一个序列，你需要处理如下两种询问。

"C a b c"表示给[a, b]区间中的值全部增加c (-10000 ≤ c ≤ 10000)。

"Q a b" 询问[a, b]区间中所有值的和。

Input

第一行包含两个整数N, Q。1 ≤ N,Q ≤ 100000.

第二行包含n个整数，表示初始的序列A (-1000000000 ≤ Ai ≤ 1000000000)。

接下来Q行询问，格式如题目描述。

Output

对于每一个Q开头的询问，你需要输出相应的答案，每个答案一行。

Sample Input

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

Sample Output

4
55
9
15

对一组数的多次更新维护用线段树，但如果用它给定的区间去依次单点更新会超时，就需要区间更新，这时引入延迟更新概念，即只是在需要更新的这一段上加上一个标记，当再次被更新时才向下传递。树若从1开始创建，左子叶为2，右子叶为3，k的左子叶为2*k，右子叶为2*k+1，若从0开始创建，k的左子叶为2*k+1，右子叶为2*k+2。

#include<stdio.h>
#define LL long long
int n,m,a[100010];
struct tree{
    int l,r;
    LL v,m;
}t[400000];//结构体树
void build(int root,int l,int r){//构建树
    t[root].l=l,t[root].r=r,t[root].m=0;
    if(l==r){
        t[root].v=a[l];
        return ;
    }
    int mid=(t[root].l+t[root].r)/2;
    build(root*2,l,mid);
    build(root*2+1,mid+1,r);
    t[root].v=t[root*2].v+t[root*2+1].v;
}
void pushup(int root){//子叶向上传递更新值
    t[root].v=t[root*2].v+t[root*2+1].v;
}
void pushdown(int root){//向下传递更新值
    if(t[root].m){
        t[root*2].m+=t[root].m;
        t[root*2].v+=t[root].m*(t[root*2].r-t[root*2].l+1);
        t[root*2+1].m+=t[root].m;
        t[root*2+1].v+=t[root].m*(t[root*2+1].r-t[root*2+1].l+1);
        t[root].m=0;
    }
}
LL query(int root,int l,int r){//查询区间和
    if(t[root].l>=l&&r>=t[root].r)  return t[root].v;
    pushdown(root);
    int mid=(t[root].l+t[root].r)/2;
    if(mid>=r){
        return query(root*2,l,r);
    }
    else if(mid<l){
        return query(root*2+1,l,r);
    }
    else{
        return query(root*2,l,mid)+query(root*2+1,mid+1,r);
    }
}
void updata(int root,int l,int r,LL add){//更新区间
    int mid=(t[root].l+t[root].r)/2;
    if(t[root].l>=l&&r>=t[root].r){
        t[root].m+=add;
        t[root].v+=(t[root].r-t[root].l+1)*add;
        return ;
    }
    pushdown(root);
    if(mid>=r){
        updata(root*2,l,r,add);
    }
    else if(mid<l){
        updata(root*2+1,l,r,add);
    }
    else{
        updata(root*2,l,mid,add);
        updata(root*2+1,mid+1,r,add);
    }
    pushup(root);
}
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 str[10];
            int l,r,x;
            scanf("%s",str);
            if(str[0]=='Q'){
                scanf("%d%d",&l,&r);
                LL ans=query(1,l,r);
                printf("%lld\n",ans);
            }
            if(str[0]=='C'){
                scanf("%d%d%lld",&l,&r,&x);
                updata(1,l,r,x);
            }
        }
    return 0;
}