【洛谷P3372】【模板】线段树 1

题目描述

如题，已知一个数列，你需要进行下面两种操作：

1.将某区间每一个数加上x

2.求出某区间每一个数的和

输入输出格式

输入格式：

第一行包含两个整数N、M，分别表示该数列数字的个数和操作的总个数。

第二行包含N个用空格分隔的整数，其中第i个数字表示数列第i项的初始值。

接下来M行每行包含3或4个整数，表示一个操作，具体如下：

操作1： 格式：1 x y k 含义：将区间[x,y]内每个数加上k

操作2： 格式：2 x y 含义：输出区间[x,y]内每个数的和

输出格式：

输出包含若干行整数，即为所有操作2的结果。

输入输出样例

输入样例#1：

5 5
1 5 4 2 3
2 2 4
1 2 3 2
2 3 4
1 1 5 1
2 1 4

输出样例#1：

11
8
20

说明

时空限制：1000ms,128M

数据规模：

对于30%的数据：N<=8，M<=10

对于70%的数据：N<=1000，M<=10000

对于100%的数据：N<=100000，M<=100000

（数据已经过加强^_^，保证在int64/long long数据范围内）

样例说明：

题解

沉迷线段树无法自拔
记下这个代码，以后省选会用
总体思想就是维护两个线段树，一个存对应区间每一个数加的值，一个存对应区间每一个数的和（不包括区间每一个数都要加的值）
更新时如果要加区间全部包含当前区间则第一个线段树对应区间+=x
否则第二个线段树+=(min(b,r)-max(a,l))*x，之后递归。
具体实现见下（PS：一定要用long long）
My Code
（调试代码感人）

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
ll dt[2000005],dtb[2000005];
int n,m;
ll itmp[500005];
void init(int k,int l,int r) {
    if(r-l==1) {
        dtb[k]=itmp[l];
        //printf("%d %d\n",l,r);
    } else {
        //printf("%d %d\n",l,r);
        int chl=k*2,chr=k*2+1;
        init(chl,l,(l+r)/2);
        init(chr,(l+r)/2,r);
        dtb[k]=dtb[chl]+dtb[chr];
        //printf("%d %d %d %d\n",dt[k],l,(l+r)/2,r);
    }
}
void add(int a,int b,ll x,int k,int l,int r) {
    if(a<=l&&r<=b){
        dt[k]+=x;   
    }else if(l<b&&a<r){
    dtb[k]+=((ll)min(b,r)-(ll)max(a,l))*x;
    int chl=k*2,chr=k*2+1;
//  printf("%d %d\n",l,r);gouliguojiashengsiyi,qiyinhuofubiquzhi
    add(a,b,x,chl,l,(l+r)/2);
    add(a,b,x,chr,(l+r)/2,r);
    }  
}
ll query(int a,int b,int k,int l,int r) {
    if(b<=l||r<=a)return 0;
    if(a<=l&&r<=b) {
        //printf("%d %d\n",k,dt[k]);
        return dt[k]*(ll)(r-l)+dtb[k];
    }  
    if(r-l!=1){
        int chl=k*2,chr=k*2+1;
        ll res=dt[k]*((ll)min(b,r)-(ll)max(a,l));
        res+=query(a,b,chl,l,(l+r)/2);
        res+=query(a,b,chr,(l+r)/2,r);

        //printf("%d %d %d %d\n",res,l,(l+r)/2,r);
        return res;
    }
}
int main() {
        memset(dt,0,sizeof(dt));
        memset(dtb,0,sizeof(dtb));
        memset(itmp,0,sizeof(itmp));
        scanf("%d%d",&n,&m);
        for(int i=1; i<=n; i++) {
            scanf( "%lld",&itmp[i]);
        }
        init(1,1,n+1);
        int tp,x,y; 
        for(int i=1;i<=m;i++) {
            scanf("%d%d",&tp,&x);
            if(tp==2) {
                printf("%lld\n",query(x,x+1,1,1,n+1));
            }
            if(tp==1) {
                ll z;
                scanf("%d%lld",&y,&z);
                add(x,y+1,z,1,1,n+1);
            }
        }
    return 0;
}