結果

問題 No.15 カタログショッピング
ユーザー pessimistpessimist
提出日時 2024-10-21 17:01:08
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 75 ms / 5,000 ms
コード長 1,277 bytes
コンパイル時間 3,801 ms
コンパイル使用メモリ 222,504 KB
実行使用メモリ 16,768 KB
最終ジャッジ日時 2024-10-21 17:01:13
合計ジャッジ時間 4,289 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,820 KB
testcase_01 AC 2 ms
6,820 KB
testcase_02 AC 2 ms
6,820 KB
testcase_03 AC 2 ms
6,820 KB
testcase_04 AC 2 ms
6,816 KB
testcase_05 AC 70 ms
16,640 KB
testcase_06 AC 75 ms
16,768 KB
testcase_07 AC 74 ms
16,640 KB
testcase_08 AC 72 ms
16,768 KB
testcase_09 AC 73 ms
16,768 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
int main(){
  cin.tie(nullptr)->sync_with_stdio(false);
  int N, S; cin >> N >> S;
  vector<int> P(N);
  for(int& x: P) cin >> x;

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

  auto f = [] (const vector<int>& A){
    const int N = A.size();
    map<int, vector<int>> mp;
    for(int mask = 0; mask < (1 << N); ++mask){
      int sum = 0;
      for(int i = 0; i < N; ++i){
        if(~mask >> i & 1) continue;
        sum += A[i];
      }
      mp[sum].emplace_back(mask);
    }
    return mp;
  };

  auto mp0 = f(P0), mp1 = f(P1);
  vector<vector<int>> ans;
  for(auto& [s0, v0]: mp0){
    auto v1 = mp1[S - s0];
    if(v1.empty()) continue;
    for(int a0: v0){
      for(int a1: v1){
        ans.emplace_back();
        auto& a = ans.back();
        for(int i = 0; i < N0; ++i) if(a0 >> i & 1){
          a.emplace_back(i + 1);
        }
        for(int i = 0; i < N1; ++i) if(a1 >> i & 1){
          a.emplace_back(N0 + i + 1);
        }
      }
    }
  }
  sort(ans.begin(), ans.end());
  for(auto& v: ans) for(int i = 0; i < v.size(); ++i){
    cout << v[i] << " \n"[i == v.size() - 1];
  }
  return 0;
}
0