結果

問題 No.198 キャンディー・ボックス２
ユーザー minamiminami
提出日時 2019-03-28 02:03:26
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 2 ms / 1,000 ms
コード長 1,735 bytes
コンパイル時間 1,708 ms
コンパイル使用メモリ 172,500 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-11 06:17:59
合計ジャッジ時間 2,670 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 2 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 1 ms
5,248 KB
testcase_06 AC 2 ms
5,248 KB
testcase_07 AC 2 ms
5,248 KB
testcase_08 AC 2 ms
5,248 KB
testcase_09 AC 2 ms
5,248 KB
testcase_10 AC 2 ms
5,248 KB
testcase_11 AC 2 ms
5,248 KB
testcase_12 AC 1 ms
5,248 KB
testcase_13 AC 2 ms
5,248 KB
testcase_14 AC 2 ms
5,248 KB
testcase_15 AC 1 ms
5,248 KB
testcase_16 AC 2 ms
5,248 KB
testcase_17 AC 2 ms
5,248 KB
testcase_18 AC 2 ms
5,248 KB
testcase_19 AC 2 ms
5,248 KB
testcase_20 AC 2 ms
5,248 KB
testcase_21 AC 2 ms
5,248 KB
testcase_22 AC 1 ms
5,248 KB
testcase_23 AC 2 ms
5,248 KB
testcase_24 AC 2 ms
5,248 KB
testcase_25 AC 1 ms
5,248 KB
testcase_26 AC 2 ms
5,248 KB
testcase_27 AC 1 ms
5,248 KB
testcase_28 AC 2 ms
5,248 KB
testcase_29 AC 2 ms
5,248 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
using namespace std;
#ifdef _DEBUG
#include "dump.hpp"
#else
#define dump(...)
#endif

#define int long long
#define rep(i,a,b) for(int i=(a);i<(b);i++)
#define rrep(i,a,b) for(int i=(b)-1;i>=(a);i--)
#define all(c) begin(c),end(c)
const int INF = sizeof(int) == sizeof(long long) ? 0x3f3f3f3f3f3f3f3fLL : 0x3f3f3f3f;
const int MOD = 1'000'000'007;
template<class T> bool chmax(T &a, const T &b) { if (a < b) { a = b; return true; } return false; }
template<class T> bool chmin(T &a, const T &b) { if (b < a) { a = b; return true; } return false; }



signed main() {
	cin.tie(0);
	ios::sync_with_stdio(false);
	int B, N; cin >> B >> N;
	vector<int> C(N); rep(i, 0, N) {
		cin >> C[i];
	}
	sort(all(C));
	dump(C);
	vector<int> sum(N + 1);
	rep(i, 0, N)sum[i + 1] = sum[i] + C[i];
	int k = INF;
	int id = -1;
	rep(i, 0, N) {
		int plus = C[i] * i - sum[i];
		int minus = sum[N] - sum[i] - C[i] * (N - i);
		dump(plus + minus);
		if (plus > B + minus || plus + minus > k)break;
		chmin(k, plus + minus);
		id = i;
	}
	dump(id);
	dump(k);
	auto g = [&](int x) {
		int plus = x * (id + 1) - sum[id + 1];
		int minus = sum[N] - sum[id + 1] - x * (N - (id + 1));
		return plus + minus;
	};
	auto f = [&](int x) {
		int plus = x * (id + 1) - sum[id + 1];
		int minus = sum[N] - sum[id + 1] - x * (N - (id + 1));
		dump(plus, minus, x);
		return plus > B + minus || plus + minus > k;
	};
	auto binarySearch = [&](int ng, int ok) {
		if (f(ng))return ng;
		while (ng + 1 < ok) {
			int m = (ng + ok) / 2;
			if (f(m))
				ok = m;
			else
				ng = m;
		}
		return ok;
	};
	if (id == N - 1) {
		cout << k << endl;
	}
	else {
		int b = binarySearch(C[id], C[id + 1]) - 1;
		dump(b);
		cout << g(b) << endl;
	}
	return 0;
}
0