結果

問題 No.15 カタログショッピング
ユーザー rogi52rogi52
提出日時 2022-10-21 01:27:24
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 69 ms / 5,000 ms
コード長 1,141 bytes
コンパイル時間 2,558 ms
コンパイル使用メモリ 221,400 KB
実行使用メモリ 16,880 KB
最終ジャッジ日時 2023-09-13 05:13:26
合計ジャッジ時間 3,285 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,380 KB
testcase_01 AC 2 ms
4,380 KB
testcase_02 AC 1 ms
4,380 KB
testcase_03 AC 2 ms
4,376 KB
testcase_04 AC 2 ms
4,376 KB
testcase_05 AC 66 ms
16,760 KB
testcase_06 AC 69 ms
16,880 KB
testcase_07 AC 68 ms
16,712 KB
testcase_08 AC 66 ms
16,696 KB
testcase_09 AC 67 ms
16,684 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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