結果

問題 No.198 キャンディー・ボックス２
ユーザー なお
提出日時 2015-12-15 17:16:56
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 2 ms / 1,000 ms
コード長 1,090 bytes
コンパイル時間 1,695 ms
コンパイル使用メモリ 159,428 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-16 09:46:33
合計ジャッジ時間 2,321 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef vector<int> VI;
typedef vector<VI> VVI;
#define REP(i, n)           for(int(i)=0;(i)<(n);++(i))
#define FOR(i, f, t)        for(int(i)=(f);(i)<(t);(++i))
#define RREP(i, n)          for(int(i)=(n)-1;(i)>=0;--(i))
const int MOD = int(1e9+7);

ll TernarySearchLowest(ll L, ll R, function<ll(ll)> F){
    ll l = L, r = R;
    while(l<r && (r-l) > 10){
        ll l2 = l+(r-l)/3, r2 = r-(r-l)/3;
        ll suml = F(l2), sumr = F(r2);
        if(sumr < suml) l = l2; else r = r2;
    }
    ll minv = LLONG_MAX, res = L;
    for(ll i = l; i <= r; i++){
        ll v = F(i);
        if(minv > v) minv = v, res = i;
    }
    return res;
}

int main(){
    ll B,N,C[11];
    cin >> B >> N;
    ll T = B;
    REP(i,N) cin >> C[i], T += C[i];

    auto f = [&](ll v)->ll{
        ll cost = 0, b = B;
        REP(i,N) b += (C[i]-v), cost += abs(C[i]-v);
        if(b < 0) cost = 1LL<<60;
        return cost;
    };
    ll res = TernarySearchLowest(0, T/N, f);

    cout << f(res) << endl;
}
0