結果
問題 | No.1099 Range Square Sum |
ユーザー | merom686 |
提出日時 | 2020-06-26 22:43:13 |
言語 | C++17 (gcc 13.2.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 155 ms / 2,000 ms |
コード長 | 3,489 bytes |
コンパイル時間 | 1,144 ms |
コンパイル使用メモリ | 91,472 KB |
実行使用メモリ | 12,384 KB |
最終ジャッジ日時 | 2023-09-18 06:23:14 |
合計ジャッジ時間 | 4,448 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge11 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
4,376 KB |
testcase_01 | AC | 2 ms
4,376 KB |
testcase_02 | AC | 1 ms
4,376 KB |
testcase_03 | AC | 2 ms
4,376 KB |
testcase_04 | AC | 1 ms
4,384 KB |
testcase_05 | AC | 2 ms
4,380 KB |
testcase_06 | AC | 2 ms
4,380 KB |
testcase_07 | AC | 2 ms
4,380 KB |
testcase_08 | AC | 2 ms
4,376 KB |
testcase_09 | AC | 1 ms
4,380 KB |
testcase_10 | AC | 2 ms
4,380 KB |
testcase_11 | AC | 2 ms
4,380 KB |
testcase_12 | AC | 3 ms
4,380 KB |
testcase_13 | AC | 2 ms
4,376 KB |
testcase_14 | AC | 2 ms
4,380 KB |
testcase_15 | AC | 2 ms
4,376 KB |
testcase_16 | AC | 2 ms
4,376 KB |
testcase_17 | AC | 2 ms
4,376 KB |
testcase_18 | AC | 2 ms
4,380 KB |
testcase_19 | AC | 3 ms
4,380 KB |
testcase_20 | AC | 3 ms
4,376 KB |
testcase_21 | AC | 153 ms
12,332 KB |
testcase_22 | AC | 155 ms
12,288 KB |
testcase_23 | AC | 153 ms
12,356 KB |
testcase_24 | AC | 154 ms
12,292 KB |
testcase_25 | AC | 153 ms
12,384 KB |
testcase_26 | AC | 117 ms
12,356 KB |
testcase_27 | AC | 117 ms
12,320 KB |
testcase_28 | AC | 118 ms
12,372 KB |
testcase_29 | AC | 118 ms
12,232 KB |
testcase_30 | AC | 117 ms
12,296 KB |
ソースコード
#include <iostream> #include <vector> #include <string> #include <algorithm> #include <cstdio> #include <cstring> #include <cmath> using namespace std; using ll = long long; struct SegmentTree { struct Lazy { static constexpr Lazy e() { return { 0 }; } bool operator==(const Lazy &o) const { return x == o.x; } Lazy &operator*=(const Lazy &o) { return *this = { x + o.x }; } ll x; }; struct Node { static constexpr Node e() { return { 0, 0 }; } Node operator*(const Node &o) const { return { v + o.v, v2 + o.v2 }; } Node &act(int w, const Lazy &o) { return *this = { v + w * o.x, v2 + v * o.x * 2 + w * o.x * o.x }; } ll v, v2; }; SegmentTree(int n) : n(n) { for (m = 2; m < n; m *= 2); t = vector<Node>(m + n + n % 2, Node::e()); z = vector<Lazy>(m, Lazy::e()); } Node &operator[](int i) { return t[m + i]; } const Node &root() const { return t[1]; } void build() { for (int i = (n - 1 + m) / 2; i > 0; i--) t[i] = t[i * 2] * t[i * 2 + 1]; } void update(int i, const Node &o) { i += m; for (int j = m; j >= 2; j /= 2) propagate(i / j, j); for (t[i] = o; i /= 2, i > 0;) t[i] = t[i * 2] * t[i * 2 + 1]; } Node query(int l, int r, int i = 1, int il = 0, int ir = -1) { if (ir < 0) ir = m; if (l <= il && ir <= r) return t[i]; propagate(i, ir - il); Node o = Node::e(); int ic = (il + ir) / 2; if (l < ic) o = o * query(l, r, i * 2 + 0, il, ic); if (ic < r) o = o * query(l, r, i * 2 + 1, ic, ir); return o; } void range_act(int l, int r, Lazy x, int i = 1, int il = 0, int ir = -1) { if (ir < 0) ir = m; if (l <= il && ir <= r) { act(i, ir - il, x); return; } propagate(i, ir - il); int ic = (il + ir) / 2; if (l < ic) range_act(l, r, x, i * 2 + 0, il, ic); if (ic < r) range_act(l, r, x, i * 2 + 1, ic, ir); t[i] = t[i * 2] * t[i * 2 + 1]; } void propagate(int i, int w) { if (z[i] == Lazy::e()) return; for (int j = 0; j < 2; j++) act(i * 2 + j, w / 2, z[i]); z[i] = Lazy::e(); } void act(int i, int w, Lazy x) { t[i].act(w, x); if (i < m) z[i] *= x; } //void print() { // for (int j = 1; j <= m; j *= 2) { // int k = min(j * 2, m + n); // for (int i = j; i < k; i++) { // if (j < m) { // cout << t[i].v << ':' << z[i] << string(m / j * 2 - 4, ' ') << " \n"[i == k - 1]; // } else { // cout << t[i].v << " \n"[i == k - 1]; // } // } // } //} vector<Node> t; vector<Lazy> z; int n, m; }; int main() { ios::sync_with_stdio(false); cin.tie(0); int n; cin >> n; SegmentTree st(n); for (int i = 0; i < n; i++) { int a; cin >> a; st[i] = { a, (ll)a * a }; } st.build(); int q; cin >> q; while (q--) { int t, l, r; cin >> t >> l >> r; l--; if (t == 1) { int x; cin >> x; st.range_act(l, r, { x }); } else { cout << st.query(l, r).v2 << '\n'; } } return 0; }