結果

問題 No.15 カタログショッピング
ユーザー nanasilinanasili
提出日時 2014-11-01 23:31:31
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 92 ms / 5,000 ms
コード長 1,972 bytes
コンパイル時間 1,263 ms
コンパイル使用メモリ 101,380 KB
実行使用メモリ 16,000 KB
最終ジャッジ日時 2024-06-09 18:41:39
合計ジャッジ時間 1,763 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
6,816 KB
testcase_01 AC 3 ms
6,944 KB
testcase_02 AC 3 ms
6,940 KB
testcase_03 AC 3 ms
6,940 KB
testcase_04 AC 2 ms
6,944 KB
testcase_05 AC 86 ms
16,000 KB
testcase_06 AC 90 ms
16,000 KB
testcase_07 AC 90 ms
15,872 KB
testcase_08 AC 92 ms
15,872 KB
testcase_09 AC 92 ms
16,000 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <vector>
#include <queue>
#include <deque>
#include <stack>
#include <map>
#include <set>
#include <algorithm>
#include <functional>
#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <ctime>
#include <cctype>
#include <string>
#include <cstring>

using namespace std;

typedef long long ll;


int main() {
  ll n, s;
  cin >> n >> s;
  ll p[n];
  for (int i = 0; i < n; i++) cin >> p[i];

  vector <int> a[66000];
  map <ll, vector <int> > memo;
  set <vector <int> > ans;
  for (ll i = 1; i < powl(2, n-(n/2)); i++) {
    ll sum = 0;
    for (ll j = 0, k = i; k != 0; k>>=1, j++) {
      if (k&1) {
        sum += p[j];
        a[i-1].push_back(j+1);
      }
    }
    if (sum == s) {
      vector <int> temp;
      for (int j = 0; j < a[i-1].size(); j++) temp.push_back(a[i-1][j]);
      ans.insert(temp);
    }
    memo[sum].push_back(i-1);
  }

  for (ll i = 1; i < powl(2, n/2); i++) {
    ll sum = 0;
    vector <int> temp;
    for (ll j = n-(n/2), k = i; k != 0; k>>=1, j++) {
      if (k&1) {
        sum += p[j];
        temp.push_back(j+1);
      }
    }
    if (memo.find(s-sum) != memo.end()) {
      for (int j = 0; j < memo[s-sum].size(); j++) {
        vector <int> temp2;
        for (int k = 0; k < a[memo[s-sum][j]].size(); k++) temp2.push_back(a[memo[s-sum][j]][k]);
        for (int k = 0; k < temp.size(); k++) temp2.push_back(temp[k]);
        ans.insert(temp2);
      }
    }else if (sum == s) {
      vector <int> temp2;
      for (int j = 0; j < temp.size(); j++) temp2.push_back(temp[j]);
      ans.insert(temp2);
    }
  }

  for (set <vector <int> >::iterator it = ans.begin(); it != ans.end(); it++) {
    for (int i = 0; i < (*it).size()-1; i++) {
      cout << (*it)[i] << " ";
    }
    cout << (*it)[(*it).size()-1];
    it++;
    if (it != ans.end() || ans.size() == 1) {
      cout << endl;
    }
    it--;
  }
}
0