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

【模板】线段树 1

题目描述

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

1. 将某区间每一个数加上 $k$。
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

样例输入 #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

提示

对于 $30\%$ 的数据：$n\le 8$，$m\le 10$。
对于 $70\%$ 的数据：$n\le {10}^3$，$m\le {10}^4$。
对于 $100\%$ 的数据：$1\le n,m\le {10}^5$。

保证任意时刻数列中任意元素的和在 $[-2^{63},2^{63})$ 内。

【样例解释】

———————————————————————————————————————————————————————

题目跳转

思路如下

套用模板即可

代码如下

/*
-----------fjn-----------
*/
#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
#define F (100000007)
#define MAXN (100+5)
typedef long long ll;
ll arr[400055];
struct node{
	int l,r;
	ll w;
	ll lg;
}; 
node ls[400055];
void js(int p,int l,int r){//建树 
	ls[p].l=l;//左 
	ls[p].r=r;//右 
	if(l==r){
		ls[p].w=arr[l];//将数据存入 
		return;
	}
	//向下分左右子树
	int mid=(l+r)/2;//对半分 
	js(p*2,l,mid);//左树 
	js(p*2+1,mid+1,r);//右树
	ls[p].w=ls[p*2].w+ls[p*2+1].w;//将左右子树的值传回给根节点 
}
void lan(int p){//懒标记下传 
	if(ls[p].lg){//懒标记不为0时候 
        ls[p*2  ].w+=ls[p].lg*(ls[p*2  ].r-ls[p*2  ].l+1);//左右段的都要加上 
        ls[p*2+1].w+=ls[p].lg*(ls[p*2+1].r-ls[p*2+1].l+1);
		ls[p*2  ].lg+=ls[p].lg;//懒标记下传 
		ls[p*2+1].lg+=ls[p].lg;
        ls[p].lg=0;
	} 
}
ll cx(int p,int x,int y){//区间查询 
	if(x<=ls[p].l&&y>=ls[p].r){//x与y包含了此区间 
		return ls[p].w;//返回这个区间的值 
	}
	lan(p);//懒标记下传 
	int mid=(ls[p].l+ls[p].r)/2;
	long long flag=0;//记录总和
	if(x<=mid) flag+=cx(p*2,x,y);
	if(y>mid ) flag+=cx(p*2+1,x,y);//加上左右儿子的值
	return flag; 
}
void xg(int p,int x,int y,ll z){//区间修改 
	if(x<=ls[p].l && y>=ls[p].r){//找到所需要修改的区间 
        ls[p].w+=ll(z)*(ls[p].r-ls[p].l+1);
        ls[p].lg+=z;
        return;
    }
    //如果不在前面区间，下放懒标记
	lan(p);
	int mid=(ls[p].l+ls[p].r)/2;
	if(x<=mid) xg(p*2,x,y,z);
	if(y>mid ) xg(p*2+1,x,y,z);//加上左右儿子的值
	ls[p].w=ls[p*2].w+ls[p*2+1].w;//区间维护 
} 
int main(){
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>arr[i];
	}
	js(1,1,n);
	for(int i=1;i<=m;i++){
//		cout<<"==="<<endl;
//		for(int i=1;i<=n*2;i++){
//			cout<<ls[i].l<<" "<<ls[i].r<<" "<<ls[i].w<<" "<<endl; 
//		}
//		cout<<"==="<<endl;
        int q,x,y;
		ll z;
        scanf("%d",&q);
        if(q==1){
            scanf("%d%d%lld",&x,&y,&z);
            xg(1,x,y,z);
        }
        else {
            scanf("%d%d",&x,&y);
            cout<<cx(1,x,y)<<endl;
        }
    }
	return 0;
}