結果

問題 No.409 ダイエット
コンテスト
ユーザー pekempey
提出日時 2016-08-06 01:06:06
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 82 ms / 2,000 ms
コード長 1,275 bytes
コンパイル時間 1,433 ms
コンパイル使用メモリ 166,384 KB
実行使用メモリ 12,076 KB
最終ジャッジ日時 2024-06-26 09:45:11
合計ジャッジ時間 6,001 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 92
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:36:42: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   36 |         for (int i = 0; i < N; i++) scanf("%lld", &D[i]);
      |                                     ~~~~~^~~~~~~~~~~~~~~

ソースコード

diff # 

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

template<class T>
struct ConvexHullTrick {
	vector<pair<T, T>> ls;

	bool need(pair<T, T> l1, pair<T, T> l2, pair<T, T> l3) {
		return -(l2.second - l1.second) * (l3.first - l2.first) < -(l3.second - l2.second) * (l2.first - l1.first);
	}

	void add(T a, T b) {
		pair<T, T> l(-a, -b);
		while (ls.size() >= 2 && !need(ls[ls.size() - 2], ls[ls.size() - 1], l)) ls.pop_back();
		ls.emplace_back(l);
	}

	T f(int k, T x) {
		return ls[k].first * x + ls[k].second;
	}

	T maximum(T x) {
		int pass = ls.size() - 1, fail = -1;
		while (pass - fail > 1) {
			int mid = (pass + fail) / 2;
			(f(mid, x) >= f(mid + 1, x) ? pass : fail) = mid;
		}
		return f(pass, x);
	}
};

int main() {
	long long N, A, B, W;
	cin >> N >> A >> B >> W;
	vector<long long> D(N);
	for (int i = 0; i < N; i++) scanf("%lld", &D[i]);

	ConvexHullTrick<long long> cht;
	vector<long long> dp(N + 1);

	for (long long i = 0; i < N; i++) {
		cht.add(-i * B, i * (i - 1) / 2 * B + A * i + dp[i]);
		dp[i + 1] = -cht.maximum(i) + i * (i + 1) / 2 * B - A * i + D[i];
	}

	long long ans = LLONG_MAX;
	for (long long i = 0; i <= N; i++) {
		long long cand = dp[i] + (N - i) * (N - i + 1) / 2 * B - A * (N - i);
		ans = min(ans, cand);
	}
	cout << W + ans << endl;
}
0