243. 一个简单的整数问题2 - 查询区间和、修改区间和

给定一个长度为 N 的数列 A，以及 M 条指令，每条指令可能是以下两种之一：

1. C l r d，表示把 A[l],A[l+1],…,A[r] 都加上 d。
2. Q l r，表示询问数列中第 l∼r 个数的和。

对于每个询问，输出一个整数表示答案。

输入格式

第一行两个整数 N,M。

第二行 N 个整数 A[i]。

接下来 M 行表示 M 条指令，每条指令的格式如题目描述所示。

输出格式

对于每个询问，输出一个整数表示答案。

每个答案占一行。

数据范围

1≤N,M≤105,
|d|≤10000,
|A[i]|≤109

输入样例：

10 5
1 2 3 4 5 6 7 8 9 10
Q 4 4
Q 1 10
Q 2 4
C 3 6 3
Q 2 4

输出样例：

4
55
9
15

 

#include<bits/stdc++.h>
#define IOS ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define endl '\n'

using namespace std;

typedef pair<int, int> PII;
typedef long long ll;

const int N = 100010;

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

inline void pushup(Node &u, Node l, Node r)
{
	u.sum = l.sum + r.sum;
}

inline void pushup(int u)
{
	pushup(tr[u], tr[u << 1], tr[u << 1 | 1]);
}

void build(int u, int l, int r)
{
	if(l == r)tr[u] = {l, r, a[l], 0};
	else
	{
		tr[u] = {l, r};
		int mid = l + r >> 1;
		build(u << 1, l, mid), build(u << 1 | 1, mid + 1, r);
		pushup(u);
	}
}

inline void pushdown(int u)
{
	Node &root = tr[u], &left = tr[u << 1], &right = tr[u << 1 | 1];
	if(root.add)
	{
		left.add += root.add, right.add += root.add;
		left.sum += root.add * (left.r - left.l + 1);
		right.sum += root.add * (right.r - right.l + 1);
		root.add = 0;
	}
}

void modify(int u, int l, int r, ll d)
{
	if(tr[u].l >= l && tr[u].r <= r)
	{
		tr[u].add += d;
		tr[u].sum += d * (tr[u]. r - tr[u].l + 1);
	}
	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;
	else
	{
		pushdown(u);
		int mid = tr[u].l + tr[u].r >> 1;
		if(l > mid)return query(u << 1 | 1, l, r);
		else if(r <= mid)return query(u << 1, l, r);
		else
		{
			ll sum = query(u << 1, l, r) + query(u << 1 | 1, l, r);
			return sum;
		}
	}
}

int main()
{
	IOS
	cin >> n >> m;
	for(int i = 1; i <= n; i ++)cin >> a[i];
	
	build(1, 1, n);
	
	char op[2];
	int l, r;
	while(m --)
	{
		cin >> op >> l >> r;
		if(*op == 'Q')
		{
			cout << query(1, l, r) << endl;
		}
		else
		{
			ll d;
			cin >> d;
			modify(1, l, r, d);
		}
	}
	
	return 0;
}

每次区间不和的时候都需要pushdown分裂一下

inline void pushdown(int u)
{
	Node &root = tr[u], &left = tr[u << 1], &right = tr[u << 1 | 1];
	if(root.add)
	{
		left.add += root.add, right.add += root.add;
		left.sum += root.add * (left.r - left.l + 1);
		right.sum += root.add * (right.r - right.l + 1);
		root.add = 0;
	}
}

注意每次操作时不能直接让left.add = root.add，因为left.add可能原本有值，要用+=root.add，

left.sum和right.sum用root.add来修改。每次pushdown完记得把root.add清空。

if(tr[u].l >= l && tr[u].r <= r)
{
	tr[u].add += d;
	tr[u].sum += d * (tr[u]. r - tr[u].l + 1);
}

还有modify操作中的add也需要注意一下。