結果

問題 No.198 キャンディー・ボックス2
ユーザー srup٩(๑`н´๑)۶
提出日時 2016-09-24 01:12:45
言語 C++11(廃止可能性あり)
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 1,153 bytes
コンパイル時間 559 ms
コンパイル使用メモリ 70,160 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-17 20:17:04
合計ジャッジ時間 1,329 ms
ジャッジサーバーID
(参考情報)
judge3 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 22 WA * 4
権限があれば一括ダウンロードができます

ソースコード

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);
	for (int i = 0; i < 50; ++i){
		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