【模板】线段树 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 的结果。

输入输出样例

输入

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

输出

11
8
20

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e5 + 10;
ll f[N];
int n, m;

struct node{
	int l, r;
	ll sum, lazy;
	node(){
		l = r = sum = lazy = 0;
	}
}a[N*4];

void update(int k)	// 更新 k 的sum 
{
	a[k].sum = a[k*2].sum + a[k*2+1].sum;
}

void build(int k, int l, int r)
{
	a[k].l = l, a[k].r = r;
	if(l == r){		// 递归到叶节点 
		a[k].sum = f[l];
		return ;
	}
	int mid = (l + r) / 2;
	build(k * 2, l, mid);
	build(k * 2 + 1, mid + 1, r);
	update(k);
}

void pushdown(int k)	// 将点 k 的懒惰标记下传 
{
	if(a[k].l == a[k].r){
		a[k].lazy = 0;
		return ;
	}
	a[k*2].sum += (a[k*2].r - a[k*2].l + 1) * a[k].lazy;
	a[k*2+1].sum += (a[k*2+1].r - a[k*2+1].l + 1) * a[k].lazy;
	a[k*2].lazy += a[k].lazy;
	a[k*2+1].lazy += a[k].lazy;
	a[k].lazy = 0; 
}

void changeSegment(int k, int l, int r, int x)	// 区间修改 
{
	if(a[k].l == l && a[k].r == r){
		a[k].sum += (r - l + 1) * x;	// 更新该区间的sum 
		a[k].lazy += x;		// 懒惰标记叠加 
		return ;
	}
	if(a[k].lazy) pushdown(k);
	int mid = (a[k].l + a[k].r) / 2;
	if(r <= mid) changeSegment(k * 2, l, r, x);		// 被修改的区间在左区间 
	else if(l > mid) changeSegment(k * 2 + 1, l, r, x);		// 被修改的区间在右区间 
	else{	// 把修改区间分解成两块，分别往左右区间递归 
		changeSegment(k * 2, l, mid, x);
		changeSegment(k * 2 + 1, mid + 1, r, x);
	}
	update(k);
}

ll query(int k, int l, int r)	// 区间查询
{
	if(a[k].lazy) pushdown(k);
	if(a[k].l == l && a[k].r == r) return a[k].sum;
	int mid = (a[k].l + a[k].r) / 2;
	if(r <= mid) return query(k * 2, l, r);
	if(l > mid) return query(k * 2 + 1, l, r);
	return query(k * 2, l, mid) + query(k * 2 + 1, mid + 1, r);	
}

int main()
{
	cin >> n >> m;
	for(int i = 1; i <= n; i++){
		cin >>f[i];
	}
	build(1, 1, n);
	for(int i = 1; i <= m; i++){
		int a, b, c, d;
		cin >> a;
		if(a == 1){
			cin >> b >> c >> d;
			changeSegment(1, b, c, d);
		}
		else if(a == 2){
			cin >> b >> c;
			ll ans = query(1, b, c);
			cout << ans << endl;
		}
	}
	return 0;
}