結果

問題 No.2918 Divide Applicants Fairly
ユーザー みうね
提出日時 2024-09-17 02:04:49
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,480 bytes
コンパイル時間 4,427 ms
コンパイル使用メモリ 286,752 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-07 11:30:42
合計ジャッジ時間 5,647 ms
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2 WA * 1
other AC * 55 WA * 4
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;

int main() {
    int N;
    cin >> N;
    vector<int> R(N);
    for (int &r : R) cin >> r;
    if (N > 22) {
        cout << 0 << endl;
        return 0;
    }
    vector<int> eq;
    vector<vector<int>> pl(N, vector<int>());
    vector<vector<int>> mi(N, vector<int>());
    function<void(int, int, int)> frontexh = [&](int i, int dif, int sum) {
        if (i == N / 2) {
            if (dif == 0)
                eq.emplace_back(sum);
            else if (dif > 0)
                pl[dif].emplace_back(sum);
            else
                mi[-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人のグループとなる場合を除去.
    eq.erase(eq.begin());
    sort(eq.begin(), eq.end());
    for (int i = 0; i < N; ++i) {
        sort(pl[i].begin(), pl[i].end());
        sort(mi[i].begin(), mi[i].end());
    }
    int ans = 1 << 30;
    for (auto &&e : eq) ans = min(ans, abs(e));
    // 後ろ半分だけを選んで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) ans = min(ans, abs(sum));
                else flag = true;
                auto it1 = lower_bound(pl[dif].begin(), pl[dif].end(), -sum);
                if (it1 != pl[dif].end()) ans = min(ans, abs(sum + *it1));
                if (it1 != pl[dif].begin()) ans = min(ans, abs(sum + *(--it1)));
            } else if (dif > 0) {
                auto it1 = lower_bound(mi[dif].begin(), mi[dif].end(), -sum);
                if (it1 != mi[dif].end()) ans = min(ans, abs(sum + *it1));
                if (it1 != mi[dif].begin()) ans = min(ans, abs(sum + *(--it1)));
            } else {
                auto it1 = lower_bound(pl[-dif].begin(), pl[-dif].end(), -sum);
                if (it1 != pl[-dif].end()) ans = min(ans, abs(sum + *it1));
                if (it1 != pl[-dif].begin()) ans = min(ans, abs(sum + *(--it1)));
            }
        } 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