1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77
| #include <bits/stdc++.h> using namespace std; typedef long long ll; const int maxn = 2e5 + 50; int n, m, a[maxn];
struct { ll s, tag; int l, r; #define l(x) tr[x].l #define r(x) tr[x].r #define s(x) tr[x].s #define tag(x) tr[x].tag #define lc(x) x * 2 #define rc(x) x * 2 + 1 } tr[4 * maxn];
void build(int p, int l, int r) { l(p) = l, r(p) = r; if (l == r) return; int mid = (l + r) / 2; build(lc(p), l, mid), build(rc(p), mid + 1, r); }
void up(int p) { s(p) = s(lc(p)) + s(rc(p)); }
void down(int p) { if (tag(p)) { s(lc(p)) += tag(p) * (r(lc(p)) - l(lc(p)) + 1); s(rc(p)) += tag(p) * (r(rc(p)) - l(rc(p)) + 1); tag(lc(p)) += tag(p), tag(rc(p)) += tag(p); tag(p) = 0; } }
void add(int p, int l, int r, int k) { if (l(p) >= l && r(p) <= r) { s(p) += k * (r(p) - l(p) + 1); tag(p) += k; return; } down(p); int mid = (l(p) + r(p)) / 2; if (l <= mid) add(lc(p), l, r, k); if (r > mid) add(rc(p), l, r, k); up(p); }
ll ask(int p, int l, int r) { if (l(p) >= l && r(p) <= r) return s(p); down(p); int mid = (l(p) + r(p)) / 2; ll ans = 0; if (l <= mid) ans += ask(lc(p), l, r); if (r > mid) ans += ask(rc(p), l, r); return ans; }
int main() { scanf("%d%d", &n, &m); for (int i = 1; i <= n; ++i) scanf("%d", &a[i]); int opt, x, y, k, d; build(1, 1, n); while (m--) { scanf("%d", &opt); if (opt == 1) { scanf("%d%d%d%d", &x, &y, &k, &d); add(1, x, x, k); if (x < y) add(1, x + 1, y, d); if (y < n) add(1, y + 1, y + 1, -(k + (y - x) * d)); } else { scanf("%d", &x); printf("%lld\n", (ll)a[x] + ask(1, 1, x)); } } return 0; }
|