线段树基础题（范围重置，维护区间和）

线段树基础题（范围重置，维护区间和）

---

题意： 维护区间和，简单区间修改（范围重置）。输入 1 为修改操作，输入 2 为查询操作。
思路： 线段树维护区间和，lazy tag维护线段树。
与上题基本一致，将区间修改改为 tree[p] = (r - l + 1) * k、lazy tag 修改改为 tag[p] = k 即可。
本题无原题链接（若有错误，感谢指正），附手写样例：

输入：
8 6
659 463 793 740 374 330 772 681 
2 1 8
1 1 8 1
2 1 8
1 4 8 10
2 4 5
2 3 4
输出：
4812
8
20
11

代码：

// LH
#include<bits/stdc++.h>
using namespace std;
#define int long long
typedef long long ll;
typedef pair<int,int> PII;
typedef pair<ll,ll> PLL;
typedef vector<int> VI;
typedef vector<ll> VL;
typedef vector<PII> VPII;
typedef vector<PLL> VPLL;
//维护区间和 ，范围重置 , lazy tag
const int N = 1e5 + 5;
int ls(int x) {return x << 1;}
int rs(int x) {return x << 1 | 1;}
int tree[N<<2] , tag[N<<2];
void up(int p) {tree[p] = tree[ls(p)] + tree[rs(p)];}//信息汇总，在修改了子区间范围的值后导致父范围的值发生改变应该将子范围的信息重新汇总到父范围
void down(int p , int l , int r) {	//传递lazy tag父范围已经改变过，当需要查询或改变子区间范围值时，将父范围的lazy tag下传，并修改子区间范围的值
	if(tag[p]) {
		int mid = l + r >> 1;
		tree[ls(p)] = tag[p] * (mid - l + 1);
		tree[rs(p)] = tag[p] * (r - mid);
		tag[ls(p)] = tag[p];
		tag[rs(p)] = tag[p];
		tag[p] = 0;
	}
}
void build(int l , int r , int p) {	//建树
	if(l == r) {
		cin >> tree[p];
		return;
	}
	int mid = l + r >> 1;
	build(l , mid , ls(p));
	build(mid + 1 , r , rs(p));
	up(p);
}
int query(int l , int r , int p , int lx , int rx) {	//查询时若需查询子区间应将lazy tag下传，并根据lazy tag修改子区间范围的值
	if(lx <= l && r <= rx) {
		return tree[p];
	}
	down(p , l , r);
	int mid = l + r >> 1;
	int sum = 0;
	if(lx <= mid) sum += query(l , mid , ls(p) , lx , rx);
	if(rx > mid) sum += query(mid+1 , r , rs(p) , lx , rx);
	return sum;
}
void update(int l , int r , int p , int lx , int rx , int k) {	//修改（此处为范围重置），在需要修改子区间范围时，记得下传lazy tag
	if(lx <= l && r <= rx) {
		tree[p] = (r - l + 1) * k;
		tag[p] = k;
		return;
	}
	down(p , l , r);		//不要忘了下传lazy tag否则可能会导致子范围未正确修改，up函数上传错误信息至父范围
	int mid = l + r >> 1;
	if(lx <= mid) update(l , mid , ls(p) , lx , rx , k);
	if(rx > mid) update(mid+1 , r , rs(p) , lx , rx , k);
	up(p);
}
void solve(){
	int n , m;
	cin >> n >> m;
	build(1 , n , 1);
	while(m --) {
		int op;
		cin >> op;
		if(op == 1) {
			int x , y , k;
			cin >> x >> y >> k;
			update(1 , n , 1 , x , y , k);
		}else {
			int x , y;
			cin >> x >> y;
			cout << query(1 , n , 1 , x , y) << '\n';
		}
	}
}
signed main() {
	std::ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	int t = 1;
	//cin >> t;
	while(t--){
		solve();
	}
	
	return 0;
}