結果

問題 No.198 キャンディー・ボックス2
ユーザー willowoooo
提出日時 2015-05-04 18:58:35
言語 C++11(廃止可能性あり)
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 892 bytes
コンパイル時間 602 ms
コンパイル使用メモリ 73,176 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-05 18:58:36
合計ジャッジ時間 1,461 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3 WA * 1
other AC * 8 WA * 18
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<iostream>
#include<algorithm>
#include<string>
#include<vector>
#include<set>
using namespace std;

void solve198()
{
	long long B, N;
	cin >> B >> N;
	long long ans = 0;
	vector<long long > BOX(N);

	long long val = 0;
	for (int i = 0; i < N; i++)
	{
		cin >> BOX[i];
		val += BOX[i];
	}

	val = val / N;
	vector<pair<long long,int>> goal(N);
	for (int i = 0; i < N; i++)
	{
		goal[i] = make_pair(abs(BOX[i] - val),i);
	}

	sort(BOX.begin(), BOX.end());
	reverse(BOX.begin(), BOX.end());
	
	sort(goal.begin(), goal.end());
	for (int i = 0; i < N; i++)
	{
		val = BOX[goal[i].second];
		for (int i = 0; i < N; i++)
		{
			if (BOX[i]>val){
				B += BOX[i] - val;
				ans += BOX[i] - val;
			}
			else if (BOX[i] < val){
				B += val - BOX[i];
				ans += val-BOX[i];
			}

		}
		if (B >= 0){
			cout << ans << endl;
		
			return;
		}
	}
	return;
}
int main(){
	solve198();
	return 0;
}
0