#1002143 (C++17) No.913 木の燃やし方

• 問題: No.913 木の燃やし方
• ユーザー: noimi
• 提出日時: 2024-08-16 16:35:38
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 329 ms / 3,000 ms
• コード長: 4,260 bytes
• コンパイル時間: 2,956 ms
• コンパイル使用メモリ: 228,564 KB
• 実行使用メモリ: 14,380 KB
• 最終ジャッジ日時: 2024-08-16 16:35:56
• 合計ジャッジ時間: 16,902 ms
• ジャッジサーバーID （参考情報）: judge5 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	5 ms; 5,376 KB
testcase_04	AC	5 ms; 5,376 KB
testcase_05	AC	5 ms; 5,376 KB
testcase_06	AC	4 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	304 ms; 10,752 KB
testcase_09	AC	317 ms; 10,368 KB
testcase_10	AC	293 ms; 10,368 KB
testcase_11	AC	289 ms; 10,624 KB
testcase_12	AC	311 ms; 11,088 KB
testcase_13	AC	295 ms; 10,624 KB
testcase_14	AC	283 ms; 10,624 KB
testcase_15	AC	281 ms; 10,368 KB
testcase_16	AC	267 ms; 10,752 KB
testcase_17	AC	257 ms; 10,496 KB
testcase_18	AC	256 ms; 10,496 KB
testcase_19	AC	230 ms; 11,648 KB
testcase_20	AC	287 ms; 10,960 KB
testcase_21	AC	277 ms; 11,096 KB
testcase_22	AC	268 ms; 11,092 KB
testcase_23	AC	245 ms; 11,016 KB
testcase_24	AC	214 ms; 11,428 KB
testcase_25	AC	168 ms; 11,648 KB
testcase_26	AC	329 ms; 11,120 KB
testcase_27	AC	244 ms; 14,380 KB
testcase_28	AC	271 ms; 11,092 KB
testcase_29	AC	274 ms; 11,072 KB
testcase_30	AC	288 ms; 11,100 KB
testcase_31	AC	225 ms; 13,964 KB
testcase_32	AC	229 ms; 14,284 KB
testcase_33	AC	233 ms; 14,376 KB
testcase_34	AC	274 ms; 11,088 KB
testcase_35	AC	272 ms; 10,956 KB
testcase_36	AC	257 ms; 11,224 KB

コンパイルメッセージ

main.cpp:20:12: warning: use of 'auto' in parameter declaration only available with '-std=c++20' or '-fconcepts'
   20 | bool chmin(auto &a, auto b) { return a > b ? a = b, 1 : 0; }
      |            ^~~~
main.cpp:20:21: warning: use of 'auto' in parameter declaration only available with '-std=c++20' or '-fconcepts'
   20 | bool chmin(auto &a, auto b) { return a > b ? a = b, 1 : 0; }
      |                     ^~~~
main.cpp:21:12: warning: use of 'auto' in parameter declaration only available with '-std=c++20' or '-fconcepts'
   21 | bool chmax(auto &a, auto b) { return a < b ? a = b, 1 : 0; }
      |            ^~~~
main.cpp:21:21: warning: use of 'auto' in parameter declaration only available with '-std=c++20' or '-fconcepts'
   21 | bool chmax(auto &a, auto b) { return a < b ? a = b, 1 : 0; }
      |                     ^~~~

ソースコード

#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")

#include <bits/stdc++.h>
using namespace std;

#define ll long long
#define pii pair<int, int>
#define pll pair<ll, ll>
#define vi vector<int>
#define vl vector<ll>
#define ov3(a, b, c, name, ...) name
#define rep0(n) for(ll aaaaa = 0; aaaaa < (n); aaaaa++)
#define rep1(i, n) for(ll i = 0; i < (n); i++)
#define rep2(i, a, b) for(ll i = (a); i < (b); i++)
#define rep(...) ov3(__VA_ARGS__, rep2, rep1, rep0)(__VA_ARGS__)
#define fore(e, v) for(auto &&e : v)
#define all(v) begin(v), end(v)
#define si(a) (int)(size(a))
bool chmin(auto &a, auto b) { return a > b ? a = b, 1 : 0; }
bool chmax(auto &a, auto b) { return a < b ? a = b, 1 : 0; }
const int inf = 11e8;
const ll infl = 2e18;

#define i128 __int128_t

template <bool isMin = true> struct CHT {
#define x first
#define y second
    CHT() = default;
    deque<pll> v;
    bool empty() { return v.empty(); }
    void clear() { return v.clear(); }
    inline int sgn(ll x) { return !x ? 0 : (x < 0 ? -1 : 1); }
    using D = long double;
    inline bool check(const pll &a, const pll &b, const pll &c) {
        if(b.y == a.y or c.y == b.y) return sgn(b.x - a.x) * sgn(c.y - b.y) >= sgn(c.x - b.x) * sgn(b.y - a.y);
        return D(b.x - a.x) * sgn(c.y - b.y) / D(abs(b.y - a.y)) >= D(c.x - b.x) * sgn(b.y - a.y) / D(abs(c.y - b.y));
    }
    void add(ll a, ll b) {
        if(!isMin) a *= -1, b *= -1;
        pll line(a, b);
        if(empty())
            v.emplace_front(line);
        else {
            if(ll c = v[0].x; c <= a) {
                if(c == a) {
                    if(v[0].y <= b) return;
                    v.pop_front();
                }
                while(si(v) >= 2 and check(line, v[0], v[1])) v.pop_front();
                v.emplace_front(line);
            } else {
                assert(a <= v.back().x);
                if(v.back().x == a) {
                    if(v.back().y <= b) return;
                    v.pop_back();
                }
                while(si(v) >= 2 and check(v[si(v) - 2], v.back(), line)) v.pop_back();
                v.emplace_back(line);
            }
        }
    }
    ll get_y(const pll &a, const ll &x) { return a.x * x + a.y; }
    ll query(ll x) {
        assert(!empty());
        int l = -1, r = si(v) - 1;
        while(l + 1 < r) {
            int m = (l + r) >> 1;
            if(get_y(v[m], x) >= get_y(v[m + 1], x))
                l = m;
            else
                r = m;
        }
        return get_y(v[r], x) * (isMin ? 1 : -1);
    }
    ll query_monotone_inc(ll x) {
        assert(!empty());
        while(si(v) >= 2 and get_y(v[0], x) >= get_y(v[1], x)) v.pop_front();
        return get_y(v[0], x) * (isMin ? 1 : -1);
    }
    ll query_monotone_dec(ll x) {
        assert(!empty());
        while(si(v) >= 2 and get_y(v.back(), x) >= get_y(v.end()[-2], x)) v.pop_back();
        return get_y(v.back(), x) * (isMin ? 1 : -1);
    }
#undef x
#undef y
};

int main() {
    int n;
    cin >> n;
    vl a(n);
    fore(e, a) cin >> e;
    vl c(n + 1);
    rep(i, n) c[i + 1] = c[i] + a[i];
    // (r - l) ^ 2 + c[r] - c[l]
    // (r ^ 2 + c[r]) - (c[l] - l^2) - 2lr

    vl res(n, infl);
    auto rec = [&](auto &&f, int l, int r) {
        if(l + 1 == r) {
            chmin(res[l], 1 + a[l]);
            return;
        }
        int mid = l + r >> 1;
        vl lres(mid - l);
        CHT<true> cht;
        rep(i, mid, r + 1) { cht.add(-2 * i, i * i + c[i]); }
        for(ll i = mid - 1; i >= l; i--) {
            ll t = -(c[i] - i * i);
            lres[i - l] = cht.query_monotone_dec(i) + t;
        }
        rep(i, si(lres)) {
            if(i) chmin(lres[i], lres[i - 1]);
            chmin(res[l + i], lres[i]);
        }
        cht.clear();
        rep(i, l, mid + 1) { cht.add(-2 * i, -(c[i] - i * i)); }
        vl rres(r - mid + 1);
        rep(i, mid + 1, r + 1) {
            ll t = (i * i + c[i]);
            rres[i - mid] = cht.query_monotone_inc(i) + t;
        }
        for(int i = r; i > mid; i--) {
            chmin(res[i - 1], rres[i - mid]);
            chmin(rres[i - mid - 1], rres[i - mid]);
        }
        f(f, l, mid), f(f, mid, r);
    };
    rec(rec, 0, n);
    rep(i, n) { cout << res[i] << '\n'; }
}