【洛谷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;}