結果

問題 No.198 キャンディー・ボックス２
ユーザー ty70
提出日時 2015-05-28 18:23:09
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
TLE  
実行時間 -
コード長 2,326 bytes
コンパイル時間 716 ms
コンパイル使用メモリ 94,888 KB
実行使用メモリ 16,824 KB
最終ジャッジ日時 2024-07-06 12:10:45
合計ジャッジ時間 5,033 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 4
other TLE * 1 -- * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <string>
#include <stack>
#include <queue>
#include <deque>
#include <set>
#include <map>
#include <algorithm>	// require sort next_permutation count __gcd reverse etc.
#include <cstdlib>	// require abs exit atof atoi 
#include <cstdio>		// require scanf printf
#include <functional>
#include <numeric>	// require accumulate
#include <cmath>		// require fabs
#include <climits>
#include <limits>
#include <cfloat>
#include <iomanip>	// require setw
#include <sstream>	// require stringstream 
#include <cstring>	// require memset
#include <cctype>		// require tolower, toupper
#include <fstream>	// require freopen
#include <ctime>		// require srand
#define rep(i,n) for(int i=0;i<(n);i++)
#define ALL(A) A.begin(), A.end()
#define INF 1LL<<60

using namespace std;

typedef long long ll;
typedef pair<int, int> P;

ll C[12];

int N;
ll f(ll base ){
	ll curr = 0LL;
	rep (i, N ) curr += abs (C[i] - base );

	return curr;
}

ll ternary_search (ll lo, ll hi ){

	if (N == 1 ) return 0LL;

	int cnt = 1000;
	while (cnt-- ){
		ll cl = lo + (hi - lo )/3LL;
		ll ch = lo + 2LL*(hi - lo )/3LL; 
		if (f(cl) > f(ch) )
			lo = cl;
		else
			hi = ch;
	} // end while


	ll res = INF;
	for (ll base = max(lo-100LL, 0LL ); base <= hi; base++ ){
		ll curr = f(base );
		res = min (res, curr );
	} // end for

	return res;
}

ll linear_search (ll lo, ll hi, ll B ){

	if (N == 1 ) return 0LL;

	vector<ll> c(N, 0 );
	rep (i, N ) c[i] = C[i];
	sort (ALL (c ), greater<ll>() );


	ll res = INF;
	for (ll base = lo; base <= hi; base++ ){
		ll curr = 0LL;
		rep (i, N ){
			if (c[i] - base >= 0 ){
				curr += (c[i] - base );
				B += (c[i] - base);
			}else{
				if (B >= base - c[i] ){
					curr += (base - c[i] );
					B -= (base - c[i] );
				}else{
					curr = INF;
					break;
				} // end if
			} // edn if
		} // end rep
		res = min (res, curr );
	} // end for

	return res;
}

int main()
{
	memset (C, 0LL, sizeof (C ) );
	ios_base::sync_with_stdio(0);
	ll B; cin >> B;
	cin >> N;
	ll sum = B;
	rep (i, N ) cin >> C[i], sum += C[i];

	ll max_base = sum/(ll)N;

	ll lo = 0;
	ll hi = max_base + 1LL;

//	ll res1 = ternary_search (lo, hi, B );
//	cout << res1 << endl;

	ll res2 = linear_search (lo, hi, B );
	cout << res2 << endl;

//	cout << res1 << ' ' << res2 << endl;
	   
	return 0;
}
0