結果

問題 No.15 カタログショッピング
コンテスト
ユーザー 👑 CleyL
提出日時 2022-09-13 10:06:01
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 116 ms / 5,000 ms
コード長 3,069 bytes
コンパイル時間 1,864 ms
コンパイル使用メモリ 129,796 KB
最終ジャッジ日時 2025-02-07 04:44:33
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

//yukicoder@cpp17

#include <iostream>

#include <cmath>
#include <algorithm>

#include <iomanip>

#include <string>
#include <vector>
#include <set>
#include <stack>
#include <queue>
#include <map>

#include <bitset>

#include <complex>

#include <cctype>
#include <utility>
#include <climits>

#include <numeric>

using namespace std;
using ll = long long;
using P = pair<ll,ll>;

const ll MOD = 998244353;
const ll MODx = 1000000007;
const int INF = (1<<30)-1;
const ll LINF = (1LL<<62LL)-1;
const double EPS = (1e-10);

P ar4[4] = {{0,1},{0,-1},{1,0},{-1,0}};
P ar8[8] = {{-1,-1},{-1,0},{-1,1},{0,-1},{0,1},{1,-1},{1,0},{1,1}};




template <typename T> vector<T> make_vector(size_t a, T b) { return vector<T>(a, b); }
template <typename... Ts> auto make_vector(size_t a, Ts... ts) { return vector<decltype(make_vector(ts...))>(a, make_vector(ts...)); }
 
 /*
確認ポイント
cout << fixed << setprecision(n) << 小数計算//n桁の小数表記になる

計算量は変わらないが楽できるシリーズ
min(max)_element(iter,iter)で一番小さい(大きい)値のポインタが帰ってくる
count(iter,iter,int)でintがiterからiterの間にいくつあったかを取得できる
*/

/*
function corner below
*/


/*
Function corner above
*/

__attribute__((constructor))
void initial() {
 cin.tie(0);
 ios::sync_with_stdio(false);
}

int main(){
  int n;
  long long s;cin>>n>>s;
  vector<long long> A(n/2),B(n/2+(n%2));
  int divide = n/2;
  for(int i = 0; n/2 > i; i++){
    cin>>A[i];
  }
  for(int i = 0; n/2+(n%2) > i; i++){
    cin>>B[i];
  }
  map<long long,int> adex;
  vector<int> X[(1<<(n/2))+2];
  for(int i = 0; (1<<(n/2)) > i; i++){
    long long sm = 0;
    for(int j = 0; n/2 > j; j++){
      if((1<<j) & i){
        sm += A[j];
      }
    }
    if(!adex[sm]){
      adex[sm] = adex.size();
    }
    X[adex[sm]].push_back(i);
  }

  map<long long,int> bdex;
  vector<int> Y[(1<<((n/2+(n%2))))+2];
  for(int i = 0; (1<<((n/2+(n%2)))) > i; i++){
    long long sm = 0;
    for(int j = 0; (n/2+(n%2)) > j; j++){
      if((1<<j) & i){
        sm += B[j];
      }
    }
    
    if(!bdex[sm]){
      bdex[sm] = bdex.size();
    }
    Y[bdex[sm]].push_back(i);
  }
  vector<vector<int>> ret;
  for(auto x: bdex){
    if(adex[s-x.first]){
      for(int i = 0; X[adex[s-x.first]].size() > i; i++){
        for(int j = 0; Y[x.second].size() > j; j++){
          int Xnw = X[adex[s-x.first]][i];
          int Ynw = Y[x.second][j];
          vector<int> ans;
          int ind = 0;
          while(Xnw){
            if(Xnw%2){
              ans.push_back(ind+1);
            }
            Xnw/=2;
            ind++;
          }
          ind = divide;
          while(Ynw){
            if(Ynw%2){
              ans.push_back(ind+1);
            }
            Ynw/=2;
            ind++;
          }
          ret.push_back(ans);
        }
      }
    }
  }
  sort(ret.begin(),ret.end());
  for(int i = 0; ret.size() > i; i++){
    for(int j = 0; ret[i].size() > j; j++){
      cout << ret[i][j] << " \n"[j+1 == ret[i].size()];
    }
  }
}
0