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;
}
标签:int,sum,tr,mid,long,整数,242,acwing
From: https://www.cnblogs.com/itdef/p/18235427