acwing 242. 一个简单的整数问题

https://www.acwing.com/problem/content/248/

给定长度为 N 的数列 A，然后输入 M 行操作指令。

第一类指令形如 C l r d，表示把数列中第 l∼r 个数都加 d。

第二类指令形如 Q x，表示询问数列中第 x个数的值。

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

输入格式
第一行包含两个整数 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
Q 1
Q 2
C 1 6 3
Q 2
输出样例：
4
1
2
5

#include <iostream>
#include <string>

using namespace std;


const int N = 100010;
int n, q;
int a[N];


struct NODE {
	int l, r;
	long long sum, lazy;
	void update(long long x) {
		sum += 1ll * (r - l + 1) * x;
		lazy += x;
	}
}tr[4 * N];


void pushup(int u) {
	tr[u].sum = tr[u << 1].sum + tr[u << 1 | 1].sum;
}

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

void pushdown(int u) {
	int lazyv = tr[u].lazy;
	if (lazyv) {
		tr[u << 1].update(lazyv);
		tr[u << 1 | 1].update(lazyv);
		tr[u].lazy = 0;
	}
}

long long query(int u, int l, int r) {
	int L = tr[u].l; int R = tr[u].r;
	if (l <= L && r >= R) {
		return tr[u].sum;
	}
	else {
		pushdown(u);
		long long ans = 0;
		int mid = L + R >> 1;
		if (l <= mid) ans += query(u << 1, l, r);
		if (r > mid) ans += query(u << 1 | 1, l, r);
		pushup(u);
		return ans;
	}
}


void modify(int u, int l, int r, long long val) {
	int L = tr[u].l, R = tr[u].r;
	if (l <= L && r >= R) {
		tr[u].update(val);
	}
	else {
		pushdown(u);
		int mid = L + R >> 1;
		if (l <= mid) modify(u << 1, l, r, val);
		if (r > mid) modify(u << 1 | 1, l, r, val);
		pushup(u);
	}
}


int main() {
	cin >> n >> q;
	for (int i = 1; i <= n; i++) cin >> a[i];
	build(1, 1, n);

	for (int i = 1; i <= q; i++) {
		string s;
		int l, r, d, q;
		cin >> s;
		if (s == "Q") {
			cin >> l ;
			cout << query(1, l,l) << endl;
		}
		else {
			cin >> l >> r >> d;
			modify(1, l, r, d);
		}
	}

	return 0;
}

我的视频空间