結果
問題 | No.15 カタログショッピング |
ユーザー | paruki |
提出日時 | 2016-10-08 11:49:03 |
言語 | C++14 (gcc 13.2.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 52 ms / 5,000 ms |
コード長 | 1,833 bytes |
コンパイル時間 | 1,798 ms |
コンパイル使用メモリ | 186,964 KB |
実行使用メモリ | 14,772 KB |
最終ジャッジ日時 | 2023-08-14 02:29:09 |
合計ジャッジ時間 | 3,101 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge15 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
4,380 KB |
testcase_01 | AC | 1 ms
4,380 KB |
testcase_02 | AC | 2 ms
4,380 KB |
testcase_03 | AC | 3 ms
4,380 KB |
testcase_04 | AC | 1 ms
4,376 KB |
testcase_05 | AC | 52 ms
14,616 KB |
testcase_06 | AC | 51 ms
14,656 KB |
testcase_07 | AC | 52 ms
14,772 KB |
testcase_08 | AC | 52 ms
14,652 KB |
testcase_09 | AC | 51 ms
14,676 KB |
ソースコード
#include "bits/stdc++.h" using namespace std; #define FOR(i,j,k) for(int (i)=(j);(i)<(int)(k);++(i)) #define rep(i,j) FOR(i,0,j) #define each(x,y) for(auto &(x):(y)) #define mp make_pair #define all(x) (x).begin(),(x).end() #define debug(x) cout<<#x<<": "<<(x)<<endl #define smax(x,y) (x)=max((x),(y)) #define smin(x,y) (x)=min((x),(y)) #define MEM(x,y) memset((x),(y),sizeof (x)) #define sz(x) (int)(x).size() typedef long long ll; typedef pair<int, int> pii; typedef vector<int> vi; typedef vector<ll> vll; const int R = 15; int N; ll S, P[31]; vll calc(int l, int r){ int n = r - l; vll res(1ll << n); rep(mask, 1 << n){ ll sm = 0; rep(i, n)if(mask>>i&1){ sm += P[l + i]; } res[mask] = sm; } return res; } int main(){ cin >> N >> S; rep(i, N)cin >> P[i]; vector<vi> ans; if(N <= R){ vll sums = calc(0, N); rep(i, 1 << N)if(sums[i]==S){ vi a; rep(j, N)if(i >> j & 1)a.push_back(j); ans.push_back(a); } } else{ // [0, R), [R, N) vll sumsL = calc(0, R), sumsR = calc(R, N); map<ll, vi> flagsL, flagsR; rep(i, sz(sumsL))flagsL[sumsL[i]].push_back(i); rep(i, sz(sumsR))flagsR[sumsR[i]].push_back(i<<R); each(p, flagsL){ ll sumL = p.first; if(flagsR.count(S - sumL)){ auto & frs = flagsR[S-sumL]; each(fl, p.second)each(fr, frs){ vi a; rep(i, R)if(fl >> i & 1)a.push_back(i); FOR(i, R, N)if(fr >> i & 1)a.push_back(i); ans.push_back(a); } } } } sort(all(ans)); each(a, ans){ rep(i, sz(a))printf("%d%c", a[i]+1, i != sz(a) - 1 ? ' ' : '\n'); } }