結果

問題 No.913 木の燃やし方
コンテスト
ユーザー QCFium
提出日時 2020-01-23 17:52:43
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 405 ms / 3,000 ms
コード長 1,838 bytes
コンパイル時間 1,739 ms
コンパイル使用メモリ 179,832 KB
実行使用メモリ 10,548 KB
最終ジャッジ日時 2024-07-19 12:19:54
合計ジャッジ時間 17,095 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

int ri() {
	int n;
	scanf("%d", &n);
	return n;
}

// monotone one
struct CHT {
	typedef std::pair<int64_t, int64_t> Line;
	std::deque<Line> que;
	bool inv = false;
	bool needed(Line l, Line m, Line r) {
		int64_t r0 = (m.second - l.second) * (r.first - m.first);
		int64_t r1 = (r.second - m.second) * (m.first - l.first);
		if (inv) return r0 > r1;
		else return r0 < r1;
	}
	int64_t eval(Line l, int64_t x) { return x * l.first + l.second; }
	void add(Line line) {
		while (que.size() >= 2 && !needed(que[que.size() - 2], que.back(), line)) que.pop_back();
		que.push_back(line);
	}
	int64_t min(int64_t x) {
		while (que.size() >= 2 && eval(que[que.size() - 2], x) < eval(que.back(), x)) que.pop_back();
		return eval(que.back(), x);
	}
};

std::vector<int> a;
std::vector<int64_t> sum;
std::vector<int64_t> res;

void solve(int l, int r) {
	if (r - l == 1) {
		res[l] = std::min<int64_t>(res[l], a[l] + 1);
		return;
	}
	int m = l + (r - l) / 2;
	CHT cht;
	for (int i = r; i-- > m; ) cht.add({-2 * i, (int64_t) i * i + sum[i + 1]});
	int64_t cur = 1000000000000000000;
	for (int i = l; i < m; i++) {
		cur = std::min(cur, cht.min(i - 1) + (int64_t) (i - 1) * (i - 1) - sum[i]);
		res[i] = std::min(res[i], cur);
	}
	cht.que.clear();
	cht.inv = true;
	cur = 1000000000000000000;
	for (int i = l; i < m; i++) cht.add({-2 * i, (int64_t) i * i - sum[i]});
	for (int i = r; i-- > m; ) {
		cur = std::min(cur, cht.min(i + 1) + (int64_t) (i + 1) * (i + 1) + sum[i + 1]);
		res[i] = std::min(res[i], cur);
	}
	
	solve(l, m);
	solve(m, r);
}

int main() {
	int n = ri();
	a.resize(n), res.resize(n, 1000000000000000000), sum.resize(n + 1);
	for (auto &i : a) i = ri();
	for (int i = 0; i < n; i++) sum[i + 1] = sum[i] + a[i];
	solve(0, n);
	for (int i = 0; i < n; i++) std::cout << res[i] << std::endl;
	return 0;
}
0