結果

問題 No.198 キャンディー・ボックス２
コンテスト
ユーザー srup٩(๑`н´๑)۶
提出日時 2016-09-24 01:25:15
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 2 ms / 1,000 ms
コード長 1,140 bytes
コンパイル時間 806 ms
コンパイル使用メモリ 69,288 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-16 10:05:39
合計ジャッジ時間 1,747 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <cstdio>
#include <string>
#include <vector>
#include <cmath>
#include <algorithm>
using namespace std;
typedef long long ll;
#define rep(i,n) for(int i=0;i<(n);i++)
#define reps(i,f,n) for(int i=(f);i<(n);i++)
const ll INF = 1e18;

ll b;
vector<ll> c;
int n;

ll search(ll m){
	ll cnt = 0;
	rep(i, n){
		cnt += abs(m - c[i]);
	}
	return cnt;
}

int main(void){
	cin >> b >> n;
	ll sum = b;
	rep(i, n){
		ll t; cin >> t;
		c.push_back(t);
		sum += t;
		// printf("sum %lld\n", sum);
	}

	if(n == 1){
		printf("0\n");
		return 0;
	}

	sort(c.begin(), c.end());
	//下凸関数を三分探索
	// printf("sum %lld \n", sum);
	ll l = c[0];
	ll r = min(c[n - 1], sum / n) + 1;
	// ll r = c[n - 1];
	// printf("%lld %lld\n", l, r);
	while(r - l > 3){
		ll m1 = (l * 2 + r) / 3;
		ll m2 = (l + r * 2) / 3;
		// printf("%lld %lld\n", m1, m2);

		if(search(m1) > search(m2)){//左側が高い
			l = m1;
		}else{//右側が高い
			r = m2;
		}
	}
	// printf("%lld %lld\n", l, r);

	ll ans = INF;
	for (int i = l; i < r; ++i){
		// printf("%d\n", i);
		ans = min(ans, search(i));
	}
	printf("%lld\n", ans);
	return 0;
}
0