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
| #include <bits/stdc++.h> using namespace std;
#define maxn 100010 long long a[maxn]; long long tree[maxn * 4]; long long lazy[maxn * 4]; long long n, m; void pushdown(int x, int left, int right) { if (lazy[x] == 0) return; int mid = (left + right) >> 1; lazy[x << 1] += lazy[x]; tree[x << 1] += lazy[x] * (mid - left + 1); lazy[x << 1 | 1] += lazy[x]; tree[x << 1 | 1] += lazy[x] * (right - mid); lazy[x] = 0; } void build(int x, int left, int right) { if (left == right) { tree[x] = a[left]; return; } int mid = (left + right) >> 1; build(x << 1, left, mid); build(x << 1 | 1, mid + 1, right); tree[x] = tree[x << 1] + tree[x << 1 | 1]; } long long query(int x, int left, int right, int tar_left, int tar_right) { if (left == tar_left && right == tar_right) return tree[x]; pushdown(x, left, right); int mid = (left + right) >> 1; if (tar_left <= mid && tar_right > mid) { long long left_ans = query(x << 1, left, mid, tar_left, mid); long long right_ans = query(x << 1 | 1, mid + 1, right, mid + 1, tar_right); return left_ans + right_ans; } if (tar_left <= mid && tar_right <= mid) { return query(x << 1, left, mid, tar_left, tar_right); } if (tar_left >= mid) { return query(x << 1 | 1, mid + 1, right, tar_left, tar_right); } } void modify(int x, int left, int right, int tar_left, int tar_right, int y) { if (left == tar_left && right == tar_right) { tree[x] += (right - left + 1) * y; lazy[x] += y; return; } pushdown(x, left, right); int mid = (left + right) >> 1; if (tar_left <= mid && tar_right > mid) { modify(x << 1, left, mid, tar_left, mid, y); modify(x << 1 | 1, mid + 1, right, mid + 1, tar_right, y); } else if (tar_right <= mid) modify(x << 1, left, mid, tar_left, tar_right, y); else modify(x << 1 | 1, mid + 1, right, tar_left, tar_right, y); tree[x] = tree[x << 1] + tree[x << 1 | 1]; } int main() { ios::sync_with_stdio(false); cin >> n >> m; for (int i = 1; i <= n; i++) cin >> a[i]; build(1, 1, n); for (int i = 1; i <= m; i++) { int remp; cin >> remp; if (remp == 1) { int remp1, remp2, remp3; cin >> remp1 >> remp2 >> remp3; modify(1, 1, n, remp1, remp2, remp3); } else if (remp == 2) { int remp1, remp2; cin >> remp1 >> remp2; cout << query(1, 1, n, remp1, remp2) << endl; } } return 0; }
|