結果

問題 No.838 Noelちゃんと星々3
ユーザー kimiyuki
提出日時 2019-06-14 22:13:20
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,346 bytes
コンパイル時間 1,929 ms
コンパイル使用メモリ 204,568 KB
最終ジャッジ日時 2025-01-07 04:32:24
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21 WA * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define REP(i, n) for (int i = 0; (i) < (int)(n); ++ (i))
using ll = long long;
using namespace std;
template <class T, class U> inline void chmin(T & a, U const & b) { a = min<T>(a, b); }

ll solve(int n, vector<ll> & a) {
    map<ll, int> f;
    for (ll a_i : a) {
        ++ f[a_i];
    }
    vector<ll> dp(f.size() + 1, LLONG_MAX / 10);
    dp[0] = 0;
    int i = 0;
    for (auto it = f.begin(); it != f.end(); ++ it, ++ i) {
        if (i >= 1) {
            chmin(dp[i + 1], dp[i - 1] + (it->first - prev(it)->first) * it->second);
            chmin(dp[i + 1], dp[i - 1] + (it->first - prev(it)->first) * prev(it)->second);
        }
        if (i >= 2) {
            chmin(dp[i + 1], dp[i - 2] + (it->first - prev(it)->first) * prev(it)->second + (it->first - prev(prev(it))->first) * prev(prev(it))->second);
            chmin(dp[i + 1], dp[i - 2] + (it->first - prev(it)->first) * it->second + (prev(it)->first - prev(prev(it))->first) * prev(prev(it))->second);
            chmin(dp[i + 1], dp[i - 2] + (it->first - prev(prev(it))->first) * it->second + (prev(it)->first - prev(prev(it))->first) * prev(it)->second);
        }
    }
    return dp[f.size()];

}

int main() {
    int n; cin >> n;
    vector<ll> a(n);
    REP (i, n) {
        cin >> a[i];
    }
    cout << solve(n, a) << endl;
    return 0;
}
0