結果

問題 No.15 カタログショッピング
コンテスト
ユーザー Aurora
提出日時 2022-05-10 21:41:14
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 80 ms / 5,000 ms
コード長 1,385 bytes
コンパイル時間 2,206 ms
コンパイル使用メモリ 191,928 KB
実行使用メモリ 13,696 KB
最終ジャッジ日時 2024-07-18 04:30:39
合計ジャッジ時間 3,169 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 10
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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

int main(){
  long long int N,S;
  cin >> N >> S;
  int a = N/2;
  int b = N-a;
  vector<long long int> A(a),B(b);
  for(int i=0;i<a;i++) cin >> A[i];
  for(int i=0;i<b;i++) cin >> B[i];
  map<long long int,vector<vector<int>>> MapA;
  vector<vector<int>> ans(0);
  for(int i=1;i<(1<<a);i++){
    long long int Sum = 0;
    vector<int> Num(0);
    for(int j=0;j<a;j++){
      if(i & (1<<j)){
        Sum += A[j];
        Num.push_back(j+1);
      }
    }
    sort(Num.begin(),Num.end());
    if(Sum == S) ans.push_back(Num);
    else if(Sum < S) MapA[Sum].push_back(Num);
  }
  for(int i=1;i<(1<<b);i++){
    long long int Sum = 0;
    vector<int> Num(0);
    for(int j=0;j<b;j++){
      if(i & (1<<j)){
        Sum += B[j];
        Num.push_back(j+a+1);
      }
    }
    sort(Num.begin(),Num.end());
    if(Sum == S) ans.push_back(Num);
    else if(Sum < S){
      long long int rest = S-Sum;
      for(auto vec:MapA[rest]){
        vector<int> PUSH(0);
        for(int j=0;j<Num.size();j++) PUSH.push_back(Num[j]);
        for(int j=0;j<vec.size();j++) PUSH.push_back(vec[j]);
        sort(PUSH.begin(),PUSH.end());
        ans.push_back(PUSH);
      }
    }
  }
  sort(ans.begin(),ans.end());
  for(int i=0;i<ans.size();i++){
    cout << ans[i][0];
    for(int j=1;j<ans[i].size();j++) cout << " " << ans[i][j];
    cout << endl;
  }
}
0