結果

問題 No.15 カタログショッピング
コンテスト
ユーザー rogi52
提出日時 2022-10-21 01:27:24
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 69 ms / 5,000 ms
コード長 1,141 bytes
コンパイル時間 2,449 ms
コンパイル使用メモリ 214,368 KB
最終ジャッジ日時 2025-02-08 09:30:47
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 10
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
#define rep(i,n) for(int i = 0; i < (n); i++)
using namespace std;
typedef long long ll;

int main(){
    cin.tie(0);
    ios::sync_with_stdio(0);
    
    int N,S; cin >> N >> S;
    vector<int> P(N);
    rep(i,N) cin >> P[i];

    int N0 = N / 2, N1 = N - N0;
    vector<int> P0, P1;
    rep(i,N0) P0.push_back(P[i]);
    rep(i,N1) P1.push_back(P[N0 + i]);

    auto f = [&](vector<int> A) {
        int N = A.size();
        map<int,vector<int>> mp;
        rep(S,1<<N) {
            int sum = 0;
            rep(i,N) if(S & (1 << i)) sum += A[i];
            mp[sum].push_back(S);
        }
        return mp;
    };

    auto mp0 = f(P0), mp1 = f(P1);
    vector<vector<int>> ans;
    for(auto [s0, v0] : mp0) {
        auto v1 = mp1[S - s0];
        for(int a0 : v0) for(int a1 : v1) {
            vector<int> a;
            rep(i,N0) if(a0 & (1 << i)) a.push_back(i + 1);
            rep(i,N1) if(a1 & (1 << i)) a.push_back(N0 + i + 1);
            ans.push_back(a);
        }
    }

    sort(ans.begin(), ans.end());
    for(auto v : ans) rep(i,int(v.size())) cout << v[i] << " \n"[i == int(v.size()) - 1];
}
0