#503842 (C++17) No.1099 Range Square Sum

提出ソース

結果

問題	No.1099 Range Square Sum
ユーザー	しの
提出日時	2020-06-27 01:55:36
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	242 ms / 2,000 ms
コード長	5,343 bytes
コンパイル時間	959 ms
コンパイル使用メモリ	110,744 KB
実行使用メモリ	15,200 KB
最終ジャッジ日時	2023-09-18 14:45:17
合計ジャッジ時間	6,702 ms
ジャッジサーバーID （参考情報）	judge15 / judge12

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	74 ms 13,556 KB
testcase_01	AC	75 ms 13,556 KB
testcase_02	AC	74 ms 13,688 KB
testcase_03	AC	74 ms 13,640 KB
testcase_04	AC	75 ms 13,628 KB
testcase_05	AC	75 ms 13,656 KB
testcase_06	AC	74 ms 13,560 KB
testcase_07	AC	74 ms 13,556 KB
testcase_08	AC	74 ms 13,636 KB
testcase_09	AC	75 ms 13,584 KB
testcase_10	AC	75 ms 13,656 KB
testcase_11	AC	76 ms 13,572 KB
testcase_12	AC	75 ms 13,644 KB
testcase_13	AC	76 ms 13,604 KB
testcase_14	AC	75 ms 13,852 KB
testcase_15	AC	75 ms 13,652 KB
testcase_16	AC	75 ms 13,648 KB
testcase_17	AC	76 ms 13,836 KB
testcase_18	AC	76 ms 13,768 KB
testcase_19	AC	76 ms 13,644 KB
testcase_20	AC	76 ms 13,624 KB
testcase_21	AC	238 ms 15,128 KB
testcase_22	AC	238 ms 15,172 KB
testcase_23	AC	242 ms 15,200 KB
testcase_24	AC	240 ms 15,144 KB
testcase_25	AC	237 ms 15,152 KB
testcase_26	AC	200 ms 13,584 KB
testcase_27	AC	201 ms 13,616 KB
testcase_28	AC	199 ms 13,856 KB
testcase_29	AC	202 ms 13,588 KB
testcase_30	AC	199 ms 13,632 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#line 2 "lib/template.cpp"

#include <algorithm>
#include <bitset>
#include <cassert>
#include <cmath>
#include <cstring>
#include <functional>
#include <iostream>
#include <map>
#include <numeric>
#include <queue>
#include <set>
#include <stack>
using namespace std;
using ll = long long;
using ull = unsigned long long;
using VI = vector<int>;
using VVI = vector<vector<int>>;
using VLL = vector<ll>;
using VVLL = vector<vector<ll>>;
using VB = vector<bool>;
using PII = pair<int, int>;
using PLL = pair<ll, ll>;
constexpr int INF = 1000000007;
constexpr ll INF_LL = 1'000'000'000'000'000'007;
#define all(x) begin(x), end(x)
#define rall(x) rbegin(x), rend(x)
#define newl '\n'

// loops rep(until) / rep(var, until) / rep(var, from, until) / repr (reversed order)
#define OVERLOAD3(_1, _2, _3, name, ...) name
#define rep(...) OVERLOAD3(__VA_ARGS__, REPEAT_FROM_UNTIL, REPEAT_UNTIL, REPEAT)(__VA_ARGS__)
#define REPEAT(times) REPEAT_CNT(_repeat, __COUNTER__, times)
#define REPEAT_CNT(_repeat, cnt, times) REPEAT_CNT_CAT(_repeat, cnt, times)
#define REPEAT_CNT_CAT(_repeat, cnt, times) REPEAT_FROM_UNTIL(_repeat ## cnt, 0, times)
#define REPEAT_UNTIL(name, times) REPEAT_FROM_UNTIL(name, 0, times)
#define REPEAT_FROM_UNTIL(name, from, until) for (int name = from, name ## __until = (until); name < name ## __until; name++)
#define repr(...) OVERLOAD3(__VA_ARGS__, REPR_FROM_UNTIL, REPR_UNTIL, REPEAT)(__VA_ARGS__)
#define REPR_UNTIL(name, times) REPR_FROM_UNTIL(name, 0, times)
#define REPR_FROM_UNTIL(name, from, until) for (int name = (until)-1, name ## __from = (from); name >= name ## __from; name--)

template <typename T, typename U>
bool chmin(T& var, U x) { if (var > x) { var = x; return true; } else return false; }
template <typename T, typename U>
bool chmax(T& var, U x) { if (var < x) { var = x; return true; } else return false; }
ll power(ll e, ll t, ll mod = INF_LL) {
  ll res = 1; for (; t; t >>= 1, (e *= e) %= mod) if (t & 1) (res *= e) %= mod; return res;
}
template <typename T, typename U> T divceil(T m, U d) { return (m + d - 1) / d; }
template <typename T> vector<T> make_v(size_t a, T b) { return vector<T>(a, b); }
template <typename... Ts> auto make_v(size_t a, Ts... ts) {
  return vector<decltype(make_v(ts...))>(a, make_v(ts...));
}

// debugging stuff
#define repi(it, ds) for (auto it = ds.begin(); it != ds.end(); it++)
class DebugPrint { public: template <typename T> DebugPrint& operator <<(const T& v) {
#ifdef LOCAL
    cerr << v;
#endif
return *this; } } debugos; template <typename T> DebugPrint& operator<<(DebugPrint& os, const
vector<T>& vec) { os << "{"; for (int i = 0; i < vec.size(); i++) os << vec[i] << (i + 1 ==
vec.size() ? "" : ", "); os << "}"; return os; } template <typename T, typename U> DebugPrint&
operator<<(DebugPrint& os, const map<T, U>& map_var) { os << "{"; repi(itr, map_var) { os << *
itr; itr++; if (itr != map_var.end()) os << ", "; itr--; } os << "}"; return os; } template <
typename T> DebugPrint& operator<<(DebugPrint& os, const set<T>& set_var) { os << "{"; repi(
itr, set_var) { os << *itr; itr++; if (itr != set_var.end()) os << ", "; itr--; } os << "}";
return os; } template <typename T, typename U> DebugPrint& operator<<(DebugPrint& os, const
pair<T, U>& p) { os << "(" << p.first << ", " << p.second << ")"; return os; } void dump_func(
) { debugos << newl; } template <class Head, class... Tail> void dump_func(Head &&head, Tail
&&... tail) { debugos << head; if (sizeof...(Tail) > 0) { debugos << ", "; } dump_func(forward
<Tail>(tail)...); }
#ifdef LOCAL
#define dump(...) debugos << "  " << string(#__VA_ARGS__) << ": " << "[" << to_string(__LINE__) \
<< ":" << __FUNCTION__ << "]" << newl << "    ", dump_func(__VA_ARGS__)
#else
#define dump(...) ((void)0)
#endif


#line 2 "main.cpp"

constexpr int h = 18;
constexpr int MaxN = 1 << h;
ll sqs[MaxN*2], sum[MaxN*2], laz[MaxN*2];

void apply(int p, int i, ll k) {
    sqs[p] += k * sum[p] * 2 + ((k * k) << i);
    sum[p] += k << i;
    laz[p] += k;
}

void flush(int p) {
    repr(i, h) {
        int q = p >> i+1;
        apply(q<<1, i, laz[q]);
        apply(q<<1|1, i, laz[q]);
        laz[q] = 0;
    }
}

void build(int p) {
    rep(i, h) {
        int q = p >> i+1;
        sqs[q] = sqs[q<<1] + sqs[q<<1|1];
        sum[q] = sum[q<<1] + sum[q<<1|1];
        apply(q, i+1, laz[q]);
        laz[q] >>= 1;
    }
}

int main() {
    cin.tie(nullptr); ios_base::sync_with_stdio(false);

    int n; cin >> n;
    rep(i, n) cin >> sum[i + MaxN];
    rep(i, n) sqs[i + MaxN] = sum[i + MaxN] * sum[i + MaxN];
    rep(i, MaxN, MaxN*2) build(i);
    int q; cin >> q;
    rep(q) {
        int c, l, r; cin >> c >> l >> r;
        if (c == 1) {
            ll x; cin >> x;
            l--; r--;
            flush(l += MaxN); flush(r += MaxN);
            int l0 = l, r0 = r;
            int i;
            for (i = 0, r++; l < r; l >>= 1, r >>= 1, i++) {
                if (l & 1) apply(l++, i, x);
                if (r & 1) apply(--r, i, x);
            }
            build(l0); build(r0);
        } else {
            l--; r--;
            flush(l += MaxN); flush(r += MaxN);
            ll res = 0;
            for (r++; l < r; l >>= 1, r >>= 1) {
                if (l & 1) res += sqs[l++];
                if (r & 1) res += sqs[--r];
            }
            cout << res << newl;
        }
    }
}

yukicoder

結果

テストケース

ソースコード