結果

問題 No.2918 Divide Applicants Fairly
ユーザー みうね
提出日時 2024-10-09 09:12:24
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,889 bytes
コンパイル時間 5,940 ms
コンパイル使用メモリ 309,384 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-10-09 09:12:33
合計ジャッジ時間 7,615 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 3
other AC * 22 WA * 39
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include <bits/stdc++.h>
using namespace std;

template <class T, class S>
inline bool chmin(T &a, const S &b) {
    return (a > b ? a = b, 1 : 0);
}

int main() {
    int N;
    cin >> N;
    vector<int> R(N);
    for (int &r : R) cin >> r;
    if (N >= 23) {
        cout << 0 << endl;
        return 0;
    }
    unordered_map<int, vector<int>> mp;
    function<void(int, int, int)> frontexh = [&](int i, int dif, int sum) {
        if (i == N / 2) {
            mp[-dif].emplace_back(sum);
        } else {
            frontexh(i + 1, dif, sum);
            frontexh(i + 1, dif + 1, sum + R[i]);
            frontexh(i + 1, dif - 1, sum - R[i]);
        }
    };
    frontexh(0, 0, 0);
    // 前半分だけを選んで0人のグループとなる場合を除去.
    mp[0].erase(mp[0].begin());
    for (auto &[_, sums] : mp) {
        sort(sums.begin(), sums.end());
    }
    int ans = 1 << 30;
    for (int i = 0; i < mp[0].size(); ++i) {
    	chmin(ans, mp[0][i]);
    }
    if (ans == 0) {
    	cout << ans << endl;
    	return 0;
    }
    // 後ろ半分だけを選んで0人のグループとなる場合を除去.
    bool flag = false;
    function<void(int, int, int)> backexh = [&](int i, int dif, int sum) {
        if (i == N - N / 2) {
            if (dif == 0) {
                if (flag) chmin(ans, abs(sum));
                else flag = true;
            }
            auto it = lower_bound(mp[dif].begin(), mp[dif].end(), -sum);
            if (it != mp[dif].end()) chmin(ans, abs(sum + *it));
            if (it != mp[dif].begin()) chmin(ans, abs(sum + *(--it)));
        } else {
            backexh(i + 1, dif, sum);
            backexh(i + 1, dif + 1, sum + R[N / 2 + i]);
            backexh(i + 1, dif - 1, sum - R[N / 2 + i]);
        }
    };
    backexh(0, 0, 0);
    cout << ans << endl;
}
0