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 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108
| #include <bits/stdc++.h> using namespace std; typedef long long ll; typedef pair<int, int> pii; const int INF = 0x7fffffff; const int N = 1e5 + 5; const int mod = 1e9 + 7;
struct sgt { int l, r; ll dat, lz; } t[N * 4];
int a[N]; void build(int p, int l, int r) { t[p].l = l, t[p].r = r; if (l == r) { t[p].dat = a[l]; return; } int mid = (l + r) / 2; build(p * 2, l, mid); build(p * 2 + 1, mid + 1, r); t[p].dat = t[p * 2].dat + t[p * 2 + 1].dat; }
void change(int p, int x, int v) { if (t[p].l == t[p].r) { t[p].dat = v; return; } int mid = (t[p].l + t[p].r) / 2; if (x <= mid) change(p * 2, x, v); else change(p * 2 + 1, x, v); t[p].dat = t[p * 2].dat + t[p * 2 + 1].dat; }
void pushdown(int p) { if (t[p].lz) { t[p * 2].dat += t[p].lz * (t[p * 2].r - t[p * 2].l + 1); t[p * 2 + 1].dat += t[p].lz * (t[p * 2 + 1].r - t[p * 2 + 1].l + 1); t[p * 2].lz += t[p].lz; t[p * 2 + 1].lz += t[p].lz; t[p].lz = 0; } }
void changeSeg(int p, int l, int r, int x) { if (l <= t[p].l && t[p].r <= r) { t[p].dat += 1LL * (t[p].r - t[p].l + 1) * x; t[p].lz += x; return; } pushdown(p); int mid = (t[p].l + t[p].r) / 2; if (l <= mid) changeSeg(p * 2, l, r, x); if (r > mid) changeSeg(p * 2 + 1, l, r, x); t[p].dat = t[p * 2].dat + t[p * 2 + 1].dat; }
ll ask(int p, int l, int r) { if (l <= t[p].l && r >= t[p].r) return t[p].dat; pushdown(p); int mid = (t[p].l + t[p].r) / 2; ll res = 0; if (l <= mid) res += ask(p * 2, l, r); if (r > mid) res += ask(p * 2 + 1, l, r); return res; }
int main() { int n, m; cin >> n >> m; for (int i = 1; i <= n; i++) cin >> a[i]; build(1, 1, n); while (m--) { int id; cin >> id; if (id == 1) { int x, y, k; cin >> x >> y >> k; changeSeg(1, x, y, k); } else { int x, y; cin >> x >> y; cout << ask(1, x, y) << '\n'; } } return 0; }
|