結果

問題 No.198 キャンディー・ボックス２
コンテスト
ユーザー 古川修慈
提出日時 2023-09-20 17:52:03
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 1,000 ms
コード長 878 bytes
コンパイル時間 1,657 ms
コンパイル使用メモリ 173,516 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-07-06 12:49:46
合計ジャッジ時間 2,527 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

vector<long long> c;
int n;

long long search(long long m) {
    long long cnt = 0;
    for (int i = 0; i < n; i++) cnt += abs(m - c[i]);
    return (cnt);
}

int main() {

	long long b;
    cin >> b >> n;
	c.resize(n);
    for (int i = 0; i < n; i++) cin >> c[i];

	if (n == 1) {
		cout << 0 << endl;
        return (0);
    }
    long long sum = b;
    for (int i = 0; i < n; i++) sum += c[i];
    sort(c.begin(), c.end());
    long long tv = c[0], fv = sum / n + 1;
    while (fv - tv > 2) {
        long long mid1 = (tv * 2 + fv) / 3;
        long long mid2 = (tv + fv * 2) / 3;
        if (search(mid1) > search(mid2)) tv = mid1; //左側が高い
        else fv = mid2; //右側が高い
    }
    long long ans = 1e18;
    for (long long i = tv; i < fv; i++) ans = min(ans, search(i));
	cout << ans << endl;
	return (0);
}
0