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

• 問題: No.198 キャンディー・ボックス２
• ユーザー: minami
• 提出日時: 2019-03-28 02:01:43
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 1,715 bytes
• コンパイル時間: 1,697 ms
• コンパイル使用メモリ: 172,308 KB
• 実行使用メモリ: 5,248 KB
• 最終ジャッジ日時: 2024-10-11 06:17:55
• 合計ジャッジ時間: 2,607 ms
• ジャッジサーバーID （参考情報）: judge2 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,248 KB
testcase_02	WA	-
testcase_03	WA	-
testcase_04	WA	-
testcase_05	AC	2 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	2 ms; 5,248 KB
testcase_13	AC	2 ms; 5,248 KB
testcase_14	AC	2 ms; 5,248 KB
testcase_15	AC	2 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	2 ms; 5,248 KB
testcase_23	AC	2 ms; 5,248 KB
testcase_24	WA	-
testcase_25	AC	2 ms; 5,248 KB
testcase_26	AC	2 ms; 5,248 KB
testcase_27	AC	2 ms; 5,248 KB
testcase_28	AC	2 ms; 5,248 KB
testcase_29	AC	2 ms; 5,248 KB

ソースコード

#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)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;
}