結果

問題 No.198 キャンディー・ボックス２
ユーザー taba
提出日時 2017-05-20 05:44:36
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 912 bytes
コンパイル時間 1,224 ms
コンパイル使用メモリ 133,352 KB
最終ジャッジ日時 2025-01-05 00:26:12
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 16 WA * 10
権限があれば一括ダウンロードができます

ソースコード

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;
	for(auto &i:c){
		if(i>average){
			step+=i-average;
			b+=i-average;
			i=average;
		}
	}
	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