結果

問題 No.2918 Divide Applicants Fairly
ユーザー けいと
提出日時 2024-10-09 00:34:38
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,810 bytes
コンパイル時間 2,843 ms
コンパイル使用メモリ 180,868 KB
実行使用メモリ 25,564 KB
最終ジャッジ日時 2024-10-09 00:34:45
合計ジャッジ時間 5,364 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 2
other AC * 29 WA * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

void generate_subset_sums(const vector<int>& nums, vector<vector<ll>>& subset_sums) {
    int n = nums.size();
    subset_sums.resize(n + 1, vector<ll>());
    for(int mask = 0; mask < (1 << n); ++mask){
        int k = __builtin_popcount(mask);
        ll sum = 0;
        for(int i = 0; i < n; ++i){
            if(mask & (1 << i)){
                sum += nums[i];
            }
        }
        subset_sums[k].push_back(sum);
    }
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    int N;
    cin >> N;
    vector<int> R(N);
    for(int &x : R) cin >> x;

    int N1 = N / 2;
    int N2 = N - N1;
    vector<int> left_nums(R.begin(), R.begin() + N1);
    vector<int> right_nums(R.begin() + N1, R.end());

    vector<vector<ll>> left_subsets, right_subsets;
    generate_subset_sums(left_nums, left_subsets);
    generate_subset_sums(right_nums, right_subsets);

    ll min_diff = LLONG_MAX;

    for(int k = 1; k <= min(N/2, max(N1, N2)); ++k){
        if(k > N1 || k > N2) continue;
        const vector<ll>& left_sums = left_subsets[k];
        const vector<ll>& right_sums = right_subsets[k];
        vector<ll> sorted_right_sums = right_sums;
        sort(sorted_right_sums.begin(), sorted_right_sums.end());
        for(auto sum1 : left_sums){
            auto it = lower_bound(sorted_right_sums.begin(), sorted_right_sums.end(), sum1);
            if(it != sorted_right_sums.end()){
                ll sum2 = *it;
                min_diff = min(min_diff, abs(sum1 - sum2));
            }
            if(it != sorted_right_sums.begin()){
                it--;
                ll sum2 = *it;
                min_diff = min(min_diff, abs(sum1 - sum2));
            }
        }
    }

    cout << min_diff << "\n";
}
0