結果

問題 No.198 キャンディー・ボックス２
コンテスト
ユーザー daku9640
提出日時 2018-05-22 23:32:56
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 1,000 ms
コード長 2,075 bytes
コンパイル時間 1,764 ms
コンパイル使用メモリ 181,484 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-16 10:18:19
合計ジャッジ時間 2,597 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
struct cww{cww(){ios::sync_with_stdio(false);cin.tie(0);}}star;
#define all(c) (c).begin(), (c).end()
#define rall(c) (c).rbegin(), (c).rend()
#define vv(type, c, m, n, i) vector<vector<type>> c(m, vector<type>(n, i))
#define mat(type, c, m, n) vector<vector<type>> c(m, vector<type>(n));for(auto& r:c)for(auto& i:r)cin>>i;
#define rep(i,a,n) for(int i=(a), i##_len=(n); i<i##_len; ++i)
#define rrep(i,a,n) for(int i=(a), i##_len=(n); i>i##_len; --i)
#define each(x,y) for(auto &(x):(y))
#define var(type, ...)type __VA_ARGS__;Scan(__VA_ARGS__);
template<typename T> void Scan(T& t) { cin >> t; }
template<typename First, typename...Rest>void Scan(First& first, Rest&...rest) { cin >> first; Scan(rest...); }
#define vec(type, c, n) vector<type> c(n);for(auto& i:c) cin>>i;
#define len(x) ((int)(x).size())
#define mp make_pair
#define eb emplace_back
#define em emplace
#define pb pop_back
#define fi first
#define se second
#define get(a, i) get<i>(a)
typedef long long ll;
typedef long double ld;
typedef vector<int> vi;
typedef vector<double> vd;
typedef vector<long long> vll;
typedef vector<string> vs;
typedef vector<bool> vb;
typedef pair<int, int> pii;
#define mini min_element
#define maxi max_element
#define sum(a, b) accumulate(a, b)
#define chmax(a,b) if(a<b)a=b
#define chmin(a,b) if(a>b)a=b
template <class T> inline void print(T t){cout << t << "\n";}
template <class H, class... T> inline void print(H h, T... t){cout << h << " ";print(t...);}
unordered_map<ll, ll> m;
ll f(vll& v, ll& avg) {
	if (m[avg] == 0)
		each(a, v)
			m[avg] += abs(a - avg);
	return m[avg];
}
ll solve() {
	var(ll, b, n);
	vec(ll, c, n);
	if (len(set<ll>(all(c))) == 1)
		return 0;
	ll max_b = (b + sum(all(c), 0LL)) / n;
	ll low = 0, high = max_b, m1 = 1, m2 = max_b - 1;
	while (low + 3 < high) {
		m1 = (low * 2 + high) / 3;
		m2 = (low  + high * 2) / 3;
		if (f(c, m1) < f(c, m2)) {
			high = m2;
		} else {
			low = m1;
		}
	}
	return min({f(c, low), f(c, m1), f(c, m2), f(c, high)});
}
int main() {
	print(solve());
	return 0;
}
0