結果

問題 No.913 木の燃やし方
ユーザー 0w1
提出日時 2019-11-30 17:17:26
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,123 ms / 3,000 ms
コード長 2,126 bytes
コンパイル時間 2,644 ms
コンパイル使用メモリ 203,716 KB
最終ジャッジ日時 2025-01-08 06:29:27
ジャッジサーバーID
(参考情報)
judge3 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include <bits/stdc++.h>
using namespace std;
struct LineFunc {
mutable int64_t m, a, p;
bool operator<(const LineFunc& l) const { return m < l.m; }
bool operator<(int64_t x) const { return p < x; }
int64_t eval(int64_t x) const { return m * x + a; }
};
struct LineContainer : multiset<LineFunc, less<>> {
const int64_t inf = numeric_limits<int64_t>::max();
int64_t fdiv(int64_t a, int64_t b) { return a / b - ((a ^ b) < 0 && a % b); }
bool isect(iterator x, iterator y) {
if (y == end()) return x->p = inf, false;
if (x->m == y->m) x->p = x->a > y->a ? inf : -inf;
else x->p = fdiv(y->a - x->a, x->m - y->m);
return x->p >= y->p;
}
void add(int64_t m, int64_t a) {
auto z = insert({ m, a, 0 }), y = z++, x = y;
while (isect(y, z)) z = erase(z);
if (x != begin() && isect(--x, y)) isect(x, y = erase(y));
while ((y = x) != begin() && (--x)->p >= y->p) isect(x, erase(y));
}
int64_t query(int64_t x) { return lower_bound(x)->eval(x); }
};
int main() {
ios::sync_with_stdio(false);
int N;
cin >> N;
vector<int> A(N);
for (int i = 0; i < N; ++i) {
cin >> A[i];
}
vector<int64_t> ans(N, (int64_t) 1 << 60);
for (int _ = 0; _ < 2; ++_) {
vector<int64_t> pa(N + 1);
for (int i = 0; i < N; ++i) {
pa[i + 1] = pa[i] + A[i];
}
function<void(int, int)> dvcq = [&](int lb, int rb) {
if (rb - lb == 1) {
ans[lb] = min<int64_t>(ans[lb], 1 + A[lb]);
return;
}
int mb = lb + rb + _ >> 1;
LineContainer lcn;
for (int i = mb + 1; i <= rb; ++i) lcn.add(2 * i, -(pa[i] + (int64_t) i * i));
vector<int64_t> best(mb - lb, (int64_t) 1 << 60);
for (int i = lb; i < mb; ++i) best[i - lb] = -lcn.query(i) + (int64_t) i * i - pa[i];
for (int i = 1; i < mb - lb; ++i) best[i] = min(best[i], best[i - 1]);
for (int i = lb; i < mb; ++i) ans[i] = min(ans[i], best[i - lb]);
dvcq(lb, mb);
dvcq(mb, rb);
};
dvcq(0, N);
reverse(A.begin(), A.end());
reverse(ans.begin(), ans.end());
}
for (int i = 0; i < N; ++i) cout << ans[i] << endl;
return 0;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0