A Simple Problem with Integers(C++)

 

【题目描述】

这是一道模板题。

给定数列 a[1],a[2],…,a[n] ，你需要依次进行q 个操作，操作有两类：

C、lrx ：给定 l,r,x ，对于所有 i∈[l,r] ，将 a[i] 加上 x （换言之，将 a[l],a[l+1],…,a[r] 分别加上 x ）；

Q、lr ：给定l,r ，求 ∑ri=la[i] 的值（换言之，求 a[l]+a[l+1]+⋯+a[r] 的值）。

【输入】

第一行包含2 个正整数 n,q ，表示数列长度和询问个数。保证 1≤n,q≤106 。

第二行n 个整数a[1],a[2],…,a[n] ，表示初始数列。保证 ∣∣a[i]∣≤106 。

接下来 q 行，每行一个操作，为以下两种之一：

C、lrx ：对于所有 i∈[l,r] ，将 a[i] 加上 x ；

Q、lr ：输出 ∑ri=la[i] 的值。

保证 1≤l≤r≤n,∣x∣≤106 。

【输出】

对于每个 Clr 操作，输出一行，每行有一个整数，表示所求的结果。

【输入样例】

5 10
2 6 6 1 1
2 1 4
1 2 5 10
2 1 3
2 2 3
1 2 2 8
1 2 3 7
1 4 4 10
2 1 2
1 4 5 6
2 3 4

【输出样例】

15
34
32
33
50

【提示】

数据范围与提示：

对于所有数据，1≤n,q≤106,∣a[i]∣≤106,1≤l≤r≤n,∣x∣≤106 。

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5 + 5;
long long add[4 * N], sum[4 * N], n, qp, a[N];

void p(long long k, long long l, long long r) {
	if (add[k] == 0) return;
	long long mid = (l + r) >> 1;
	sum[2 * k] += (mid - l + 1) * add[k];
	add[2 * k] += add[k];
	sum[2 * k + 1] += (r - mid) * add[k];
	add[2 * k + 1] += add[k];
	add[k] = 0;
	return;
}
void m(long long k, long long l, long long r, long long x, long long y, long long v) {
	if (y < l || x > r) return;
	if (l >= x && r <= y) {
		add[k] += v;
		sum[k] += (r - l + 1) * v;
		return;
	}
	long long mid = l+r>>1;
	p(k, l, r);
	m(k * 2, l, mid, x, y, v);
	m(k * 2 + 1, mid + 1, r, x, y, v);
	sum[k] = sum[k * 2] + sum[k * 2 + 1];
}
void build(long long k, long long l, long long r) {
	if (l == r) {
		sum[k] = a[l];
		return;
	}
	long long mid = (l + r) / 2;
	build(k * 2, l, mid);
	build(k * 2 + 1, mid + 1, r);
	sum[k] = sum[k * 2] + sum[k * 2 + 1];
}
long long q(long long k, long long l, long long r, long long x, long long y) {
	if (y < l || x > r)return 0;
	if (l >= x && r <= y) return sum[k];
	long long mid = (l + r) / 2, yjc = 0;
	p(k, l, r);
	yjc += q(k * 2, l, mid, x, y) + q(k * 2 + 1, mid + 1, r, x, y);
	return yjc;
}


int main() {
	scanf("%lld%lld", &n, &qp);
	for (long long i = 1; i <= n; i++)cin >> a[i];
	build(1, 1, n);
	for (long long i = 1; i <= qp; i++) {
		char k;
		cin >> k;
		if (k == 'C') {
			long long a, b, c;
			cin >> a >> b >> c;
			m(1, 1, n, a, b, c);
		} else {
			long long a, b;
			cin >> a >> b;
			printf("%lld\n", q(1, 1, n, a, b));
		}
	}

	return 0;
}