#420879 (C++14) No.913 木の燃やし方

• 問題: No.913 木の燃やし方
• ユーザー: QCFium
• 提出日時: 2020-01-23 17:43:03
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 1,738 bytes
• コンパイル時間: 1,936 ms
• コンパイル使用メモリ: 180,080 KB
• 実行使用メモリ: 8,832 KB
• 最終ジャッジ日時: 2024-07-19 11:47:27
• 合計ジャッジ時間: 16,626 ms
• ジャッジサーバーID （参考情報）: judge1 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	1 ms; 5,376 KB
testcase_03	WA	-
testcase_04	WA	-
testcase_05	WA	-
testcase_06	WA	-
testcase_07	WA	-
testcase_08	WA	-
testcase_09	WA	-
testcase_10	WA	-
testcase_11	WA	-
testcase_12	WA	-
testcase_13	AC	338 ms; 7,040 KB
testcase_14	AC	320 ms; 7,040 KB
testcase_15	AC	322 ms; 6,784 KB
testcase_16	WA	-
testcase_17	WA	-
testcase_18	WA	-
testcase_19	AC	332 ms; 8,832 KB
testcase_20	AC	360 ms; 7,424 KB
testcase_21	WA	-
testcase_22	WA	-
testcase_23	WA	-
testcase_24	AC	324 ms; 8,192 KB
testcase_25	AC	324 ms; 8,832 KB
testcase_26	AC	370 ms; 7,296 KB
testcase_27	AC	336 ms; 8,832 KB
testcase_28	WA	-
testcase_29	WA	-
testcase_30	WA	-
testcase_31	WA	-
testcase_32	WA	-
testcase_33	WA	-
testcase_34	WA	-
testcase_35	WA	-
testcase_36	AC	347 ms; 7,296 KB

ソースコード

#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 needed(Line l, Line m, Line r) {
		return (m.second - l.second) * (r.first - m.first) <
			   (r.second - m.second) * (m.first - l.first);
	}
	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();
	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;
}