#55091 (C++11) No.198 キャンディー・ボックス２

• 問題: No.198 キャンディー・ボックス２
• ユーザー: hanorver
• 提出日時: 2015-10-23 15:54:53
• 言語: C++11 (gcc 11.4.0)
• 結果: WA
• 実行時間: -
• コード長: 900 bytes
• コンパイル時間: 684 ms
• コンパイル使用メモリ: 59,604 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-07-22 11:32:42
• 合計ジャッジ時間: 7,353 ms
• ジャッジサーバーID （参考情報）: judge2 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	WA	-
testcase_01	WA	-
testcase_02	WA	-
testcase_03	WA	-
testcase_04	TLE	-
testcase_05	WA	-
testcase_06	WA	-
testcase_07	WA	-
testcase_08	WA	-
testcase_09	AC	2 ms; 5,376 KB
testcase_10	AC	2 ms; 5,376 KB
testcase_11	AC	2 ms; 5,376 KB
testcase_12	AC	1 ms; 5,376 KB
testcase_13	WA	-
testcase_14	AC	2 ms; 5,376 KB
testcase_15	AC	2 ms; 5,376 KB
testcase_16	AC	2 ms; 5,376 KB
testcase_17	AC	2 ms; 5,376 KB
testcase_18	AC	1 ms; 5,376 KB
testcase_19	AC	2 ms; 5,376 KB
testcase_20	WA	-
testcase_21	WA	-
testcase_22	WA	-
testcase_23	WA	-
testcase_24	TLE	-
testcase_25	WA	-
testcase_26	WA	-
testcase_27	WA	-
testcase_28	WA	-
testcase_29	WA	-

ソースコード

#include<iostream>
#include<vector>
#include<numeric>

int main() {
	int box_num, having_candy_num;
	std::cin >> having_candy_num >> box_num;

	std::vector<long long> boxes(box_num);
	for (int i = 0; i < box_num; i++) std::cin >> boxes[i];

	long long ave = std::accumulate(boxes.begin(), boxes.end(), 0) / box_num;

	long long count = 0;
	for (size_t i = 0; i < boxes.size(); i++) {
		if (boxes[i] > ave) {
			for (size_t j = 0; j < boxes.size(); j++) {
				while(boxes[i] != ave && boxes[j] < ave){
					boxes[i]--;
					boxes[j]++;
					count += 2;
				}
			}
			if (boxes[i] != ave) count += boxes[i] - ave;
		} else if(boxes[i] < ave) {
			for (size_t j = 0; j < boxes.size(); j++) {
				while (boxes[i] != ave && boxes[j] > ave) {
					boxes[i]++;
					boxes[j]--;
					count+=2;
				}
			}
			if (boxes[i] != ave) count += ave - boxes[i];
		}
	}

	std::cout << count << std::endl;
	return 0;
}