线段树区间修改区间求和

Posted 2021-09-06 行码棋

题目链接

思路：

使用线段树区间修改（懒标记）
注意pushdown操作的位置，懒标记是当前节点的子节点的标记，并不包括自己这个节点。
询问的时候需要把懒标记向下更新，因为懒标记不包括父节点。

要注意pushup和pushdown的位置

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e5+5;

int n,m;
int w[N];
struct node
{
	int l,r;
	ll sum,add;
}tr[N*4];

void pushup(int u)
{
	tr[u].sum = tr[u<<1].sum + tr[u<<1|1].sum;
}
void pushdown(int u)
{
	auto &root = tr[u],&left = tr[u<<1],&right = tr[u<<1|1];
	if(root.add)
	{
		left.add += root.add,left.sum += (ll)(left.r-left.l+1)*root.add;
		right.add += root.add,right.sum += (ll)(right.r-right.l+1)*root.add;
		root.add = 0;
	}
}
void build(int u,int l,int r)
{
	if(l==r) tr[u] = {l,r,w[r],0};
	else
	{
		tr[u] = {l,r};
		int mid = tr[u].l + tr[u].r >> 1;
		build(u<<1,l,mid),build(u<<1|1,mid+1,r);
		pushup(u);
	}
}
void modify(int u,int l,int r,int d)
{
	if(tr[u].l >= l && tr[u].r <= r)
	{
		tr[u].sum += (ll)(tr[u].r - tr[u].l + 1) * d;
		tr[u].add += d;
	}
	else
	{
		pushdown(u);
		int mid = tr[u].l + tr[u].r >> 1;
		if(l <= mid) modify(u<<1,l,r,d);
		if(r > mid) modify(u<<1|1,l,r,d);
		pushup(u);
	}
}
ll query(int u,int l,int r)
{
	if(tr[u].l >= l && tr[u].r <= r) return tr[u].sum;
	
	pushdown(u); 
	int mid = tr[u].l + tr[u].r >> 1;
	ll sum = 0;
	if(l<=mid) sum = query(u<<1,l,r);
	if(r>mid) sum += query(u<<1|1,l,r);
	return sum;
}
int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++) cin>>w[i];
	build(1,1,n);
	
	char op[2];
	int l,r,d;
	while(m--)
	{
		cin>>op>>l>>r;
		if(*op=='Q') cout<<query(1,l,r)<<'\\n';
		else 
		{
			cin>>d;
			modify(1,l,r,d);
		}
	}
	return 0;
 }