Codevs_P1082 线段树练习3(线段树)

题目描述 Description
给你N个数，有两种操作：
1：给区间[a,b]的所有数增加X
2：询问区间[a,b]的数的和。

输入描述 Input Description
第一行一个正整数n，接下来n行n个整数，
再接下来一个正整数Q，每行表示操作的个数，
如果第一个数是1，后接3个正整数，
表示在区间[a,b]内每个数增加X,如果是2，
表示操作2询问区间[a,b]的和是多少。

输出描述 Output Description
对于每个询问输出一行一个答案

样例输入 Sample Input
3
1
2
3
2
1 2 3 2
2 2 3

样例输出 Sample Output
9

数据范围及提示 Data Size & Hint
数据范围
1<=n<=200000
1<=q<=200000

指针操作较慢，用f[]数组减少操作

#include<cstdio>
#include<iostream>
using namespace std;
#define N 200005
long long n,m,k,x,y,z;
long long d[N*3],f[N*3];
inline long long in(){
    long long x=0;char ch=getchar();
    while(ch>'9'||ch<'0') ch=getchar();
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x;
}
void build(long long o,long long l,long long r){
    if(l<r){
        long long m=(l+r)>>1;
        build(o*2,l,m);
        build(o*2+1,m+1,r);
        d[o]=d[o*2]+d[o*2+1];
    }
    else  d[o]=in();
}
void change(long long o,long long l,long long r,long long data){
    if(x<=l&&r<=y){
        f[o]+=data;
        d[o]+=data*(r-l+1);
        return;
    }
    long long m=(l+r)>>1;
    if(x<=m) change(o*2,l,m,data);
    if(m<y) change(o*2+1,m+1,r,data);
    d[o]=d[o*2]+d[o*2+1]+f[o]*(r-l+1);
    return;
}
long long sum(long long o,long long l,long long r,long long v){
    long long m=(l+r)>>1,res=0;
    if(x<=l&&r<=y) return d[o]+v*(r-l+1);
    if(x<=m) res+=sum(o*2,l,m,f[o]+v);
    if(y>m) res+=sum(o*2+1,m+1,r,f[o]+v);
    return res;
}
int main(){
    n=in();
    build(1,1,n);
    m=in();
    for(int i=1;i<=m;i++){
        k=in();
        if(k==1){
            x=in(),y=in(),z=in();
            change(1,1,n,z);
        }
        else{
            x=in(),y=in();
            printf("%lld\n",sum(1,1,n,0));
        }
    }
    return 0;
}