結果

問題 No.198 キャンディー・ボックス２
コンテスト
ユーザー taba
提出日時 2017-05-20 05:48:18
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 991 bytes
コンパイル時間 1,324 ms
コンパイル使用メモリ 122,112 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-12-26 07:32:40
合計ジャッジ時間 2,401 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 24 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cstdio>
#include <cstring>
#include <cmath>
#include <cassert>
#include <random>
#include <vector>
#include <algorithm>
#include <array>
#include <functional>
#include <utility>
#include <regex>
#include <tuple>
#include <map>
#include <set>
#include <iostream>

using namespace std;

int main(){
	int b,n;
	-scanf("%d%d",&b,&n);
	vector<int> c(n);
	int64_t sum=b,average,step=0,minstep=0x7fffffffffffffffL;
	for(auto &i:c){
		-scanf("%d",&i);
		sum+=i;
	}
	average=sum/n;
	sort(c.begin(),c.end());
	for(auto &i:c){
		if(i>average){
			step+=i-average;
			b+=i-average;
			i=average;
		}
	}
	sum=0;
	for(auto i:c){
		sum+=i;
	}
	average=sum/n;
	auto calcstep=[](vector<int> t,int target,int b){
		int64_t step=0;
		for(int i:t){
			step+=abs(target-i);
			b+=target-i;
		}
		return b>=0?step:0x7fffffffffffffffL;
	};
	auto t=calcstep(c,c[c.size()/2],b);
	minstep=min(minstep,t);
	t=calcstep(c,c[c.size()/2+1],b);
	minstep=min(minstep,t);
	printf("%ld\n",step+minstep);
	return 0;
}
0