結果
問題 | No.1099 Range Square Sum |
ユーザー |
![]() |
提出日時 | 2023-08-08 23:32:43 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 140 ms / 2,000 ms |
コード長 | 10,089 bytes |
コンパイル時間 | 2,087 ms |
コンパイル使用メモリ | 206,912 KB |
最終ジャッジ日時 | 2025-02-16 00:13:23 |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 30 |
ソースコード
#pragma region satashun// #pragma GCC optimize("Ofast")// #pragma GCC optimize("unroll-loops")#include <bits/stdc++.h>using namespace std;using uint = unsigned int;using ll = long long;using ull = unsigned long long;using pii = pair<int, int>;template <class T>using V = vector<T>;template <class T>using VV = V<V<T>>;template <class T>V<T> make_vec(size_t a) {return V<T>(a);}template <class T, class... Ts>auto make_vec(size_t a, Ts... ts) {return V<decltype(make_vec<T>(ts...))>(a, make_vec<T>(ts...));}template <typename T, typename V>void fill_vec(T& v, const V& val) {v = val;}template <typename T, typename V>void fill_vec(vector<T>& vec, const V& val) {for (auto& v : vec) fill_vec(v, val);}#define pb push_back#define eb emplace_back#define mp make_pair#define fi first#define se second#define rep(i, n) rep2(i, 0, n)#define rep2(i, m, n) for (int i = m; i < (n); i++)#define per(i, b) per2(i, 0, b)#define per2(i, a, b) for (int i = int(b) - 1; i >= int(a); i--)#define ALL(c) (c).begin(), (c).end()#define SZ(x) ((int)(x).size())constexpr ll TEN(int n) { return (n == 0) ? 1 : 10 * TEN(n - 1); }template <class T, class U>void chmin(T& t, const U& u) {if (t > u) t = u;}template <class T, class U>void chmax(T& t, const U& u) {if (t < u) t = u;}template <typename T>int arglb(const V<T>& v, const T& x) {return distance(v.begin(), lower_bound(ALL(v), x));}template <typename T>int argub(const V<T>& v, const T& x) {return distance(v.begin(), upper_bound(ALL(v), x));}template <class T>void mkuni(vector<T>& v) {sort(ALL(v));v.erase(unique(ALL(v)), end(v));}template <class T>vector<int> sort_by(const vector<T>& v, bool increasing = true) {vector<int> res(v.size());iota(res.begin(), res.end(), 0);if (increasing) {stable_sort(res.begin(), res.end(),[&](int i, int j) { return v[i] < v[j]; });} else {stable_sort(res.begin(), res.end(),[&](int i, int j) { return v[i] > v[j]; });}return res;}// prototypetemplate <class T, class U>ostream& operator<<(ostream& os, const pair<T, U>& p);template <class T>ostream& operator<<(ostream& os, const vector<T>& v);template <class T>istream& operator>>(istream& is, vector<T>& v);template <class T, size_t sz>ostream& operator<<(ostream& os, const array<T, sz>& arr);template <class T>ostream& operator<<(ostream& os, const set<T>& ST);template <class T>ostream& operator<<(ostream& os, const multiset<T>& ST);template <class T, class U>ostream& operator<<(ostream& os, const map<T, U>& MP);ostream& operator<<(ostream& o, __int128_t x);// io functionstemplate <class T, class U>istream& operator>>(istream& is, pair<T, U>& p) {is >> p.first >> p.second;return is;}template <class T, class U>ostream& operator<<(ostream& os, const pair<T, U>& p) {os << "(" << p.first << "," << p.second << ")";return os;}template <class T>istream& operator>>(istream& is, vector<T>& v) {for (auto& x : v) {is >> x;}return is;}template <class T>ostream& operator<<(ostream& os, const vector<T>& v) {os << "{";rep(i, v.size()) {if (i) os << ",";os << v[i];}os << "}";return os;}template <class T, size_t sz>ostream& operator<<(ostream& os, const array<T, sz>& arr) {os << '[';for (auto v : arr) os << v << ',';os << ']';return os;}template <class T>ostream& operator<<(ostream& os, const set<T>& ST) {os << "{";for (auto it = ST.begin(); it != ST.end(); ++it) {if (it != ST.begin()) os << ",";os << *it;}os << "}";return os;}template <class T>ostream& operator<<(ostream& os, const multiset<T>& ST) {os << "{";for (auto it = ST.begin(); it != ST.end(); ++it) {if (it != ST.begin()) os << ",";os << *it;}os << "}";return os;}template <class T, class U>ostream& operator<<(ostream& os, const map<T, U>& MP) {for (auto it = MP.begin(); it != MP.end(); ++it) {os << "(" << it->first << ": " << it->second << ")";}return os;}string to_string(__int128_t x) {if (x == 0) return "0";string result;if (x < 0) {result += "-";x *= -1;}string t;while (x) {t.push_back('0' + x % 10);x /= 10;}reverse(t.begin(), t.end());return result + t;}ostream& operator<<(ostream& o, __int128_t x) { return o << to_string(x); }#ifdef LOCALvoid debug_out() { cerr << endl; }template <typename Head, typename... Tail>void debug_out(Head H, Tail... T) {cerr << " " << H;debug_out(T...);}#define debug(...) \cerr << __LINE__ << " [" << #__VA_ARGS__ << "]:", debug_out(__VA_ARGS__)#define dump(x) cerr << __LINE__ << " " << #x << " = " << (x) << endl#else#define debug(...) (void(0))#define dump(x) (void(0))#endiftemplate <class T>V<T>& operator+=(V<T>& vec, const T& v) {for (auto& x : vec) x += v;return vec;}template <class T>V<T>& operator-=(V<T>& vec, const T& v) {for (auto& x : vec) x -= v;return vec;}// suc : 1 = newline, 2 = spacetemplate <class T>void print(T x, int suc = 1) {cout << x;if (suc == 1)cout << "\n";else if (suc == 2)cout << " ";}template <class T>void print(const vector<T>& v, int suc = 1) {for (int i = 0; i < v.size(); ++i)print(v[i], i == int(v.size()) - 1 ? suc : 2);}template <class T>void show(T x) {print(x, 1);}template <typename Head, typename... Tail>void show(Head H, Tail... T) {print(H, 2);show(T...);}int topbit(int t) { return t == 0 ? -1 : 31 - __builtin_clz(t); }int topbit(ll t) { return t == 0 ? -1 : 63 - __builtin_clzll(t); }int botbit(int a) { return a == 0 ? 32 : __builtin_ctz(a); }int botbit(ll a) { return a == 0 ? 64 : __builtin_ctzll(a); }int popcount(int t) { return __builtin_popcount(t); }int popcount(ll t) { return __builtin_popcountll(t); }int bit_parity(int t) { return __builtin_parity(t); }int bit_parity(ll t) { return __builtin_parityll(t); }struct prepare_io {prepare_io() {cin.tie(nullptr);ios::sync_with_stdio(false);cout << fixed << setprecision(10);}} prep_io;#pragma endregion satashun// index of root = 1// T1 : array value type// T2 : action typetemplate <class U>struct segtree {using T1 = typename U::T1;using T2 = typename U::T2;int sz, H;V<T1> a;V<T2> b;segtree() { sz = H = -1; }segtree(int n) {sz = 1, H = 0;while (sz < n) {sz *= 2, ++H;}a.assign(sz * 2, U::id1());b.assign(sz * 2, U::id2());}segtree(const V<T1>& vec) {int n = vec.size();sz = 1, H = 0;while (sz < n) {sz *= 2, ++H;}a.assign(sz * 2, U::id1());b.assign(sz * 2, U::id2());for (int i = 0; i < n; ++i) {a[sz + i] = vec[i];}for (int i = sz - 1; i >= 1; --i) {a[i] = U::op11(a[(i << 1) | 0], a[(i << 1) | 1]);}}inline T1 reflect(int k) {return b[k] == U::id2() ? a[k] : U::op21(b[k], a[k]);}void up(int i) {while (i >>= 1) {a[i] = U::op11(reflect((i << 1) | 0), reflect((i << 1) | 1));}}void propagate(int p) {for (int h = H; h > 0; --h) {int i = p >> h;a[i] = reflect(i);b[i << 1 | 0] = U::op22(b[i], b[i << 1 | 0]);b[i << 1 | 1] = U::op22(b[i], b[i << 1 | 1]);b[i] = U::id2();}}void set(int p, T1 x) {p += sz;propagate(p);a[p] = x;b[p] = U::id2();up(p);}T1 get(int p) {p += sz;propagate(p);return reflect(p);}// action on [p, q)void apply(int p, int q, T2 x) {if (p >= q) return;p += sz, q += sz;propagate(p);propagate(q - 1);for (int l = p, r = q; l < r; l >>= 1, r >>= 1) {if (l & 1) b[l] = U::op22(x, b[l]), l++;if (r & 1) --r, b[r] = U::op22(x, b[r]);}up(p);up(q - 1);}T1 get(int l, int r) {if (l >= r) return U::id1();l += sz, r += sz;propagate(l);propagate(r - 1);T1 lval = U::id1(), rval = U::id1();for (; l < r; l >>= 1, r >>= 1) {if (l & 1) lval = U::op11(lval, reflect(l++));if (r & 1) rval = U::op11(reflect(--r), rval);}return U::op11(lval, rval);}};array<ll, 3> emp;struct U {// number of 1, allusing T1 = array<ll, 3>;using T2 = ll;static T1 id1() { return emp; }static T2 id2() { return 0ll; }static T1 op11(const T1& a, const T1& b) {T1 res;res[0] = a[0] + b[0];res[1] = a[1] + b[1];res[2] = a[2] + b[2];return res;}static T1 op21(const T2& b, const T1& a) {T1 res;res[0] = a[0];res[1] = a[1] + b * a[0];res[2] = a[2] + 2 * b * a[1] + b * b * a[0];return res;}// a : new!static T2 op22(const T2& a, const T2& b) { return a + b; }};void slv() {// inputint N;cin >> N;V<array<ll, 3>> vec;rep(i, N) {ll x;cin >> x;array<ll, 3> a;a[0] = 1;a[1] = x;a[2] = x * x;vec.pb(a);}segtree<U> seg(vec);// solveint Q;cin >> Q;while (Q--) {int t, l, r;cin >> t >> l >> r;--l;if (t == 1) {ll x;cin >> x;seg.apply(l, r, x);} else {auto res = seg.get(l, r);show(res[2]);}}}int main() {int cases = 1;// cin >> cases;rep(i, cases) slv();return 0;}