結果

問題 No.15 カタログショッピング
ユーザー srup٩(๑`н´๑)۶srup٩(๑`н´๑)۶
提出日時 2016-11-08 22:50:44
言語 C++11
(gcc 11.4.0)
結果
WA  
実行時間 -
コード長 2,187 bytes
コンパイル時間 1,802 ms
コンパイル使用メモリ 80,436 KB
実行使用メモリ 5,400 KB
最終ジャッジ日時 2023-08-16 15:29:28
合計ジャッジ時間 3,527 ms
ジャッジサーバーID
(参考情報)
judge13 / judge15
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 1 ms
4,380 KB
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <vector>
#include <cstdio>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef vector<int> vint;
typedef pair<int,int> pint;
typedef vector<pint> vpint;
#define rep(i,n) for(int i=0;i<(n);i++)
#define reps(i,f,n) for(int i=(f);i<(n);i++)
#define each(it,v) for(__typeof((v).begin()) it=(v).begin();it!=(v).end();it++)
#define all(v) (v).begin(),(v).end()
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define chmax(a, b) a = (((a)<(b)) ? (b) : (a))
#define chmin(a, b) a = (((a)>(b)) ? (b) : (a))
const int MOD = 1e9 + 7;
const int INF = 1e9;

int n;
ll s;
ll p[40];
vector<pair<int, ll> > bit1, bit2;

int main(void){
	cin >> n >> s;
	rep(i, n) cin >> p[i];

	int mid1, mid2;
	mid1 = n / 2 + n % 2;
	mid2 = n / 2;
	for (int mask = 0; mask < (1 << mid1); ++mask){
		ll sum = 0;
		for (int pos = 0; pos < mid1; ++pos){
			if((mask & (1 << pos))){
				sum += p[pos];
			}
		}
		bit1.pb(mp(mask, sum));
	}
	for (int mask = 0; mask < (1 << mid2); ++mask){
		ll sum = 0;
		for (int pos = 0; pos < mid2; ++pos){
			if((mask & (1 << pos))){
				sum += p[mid1 + pos];
			}
		}
		bit2.pb(mp(mask, sum));
	}

	sort(all(bit2));
	/*
	rep(i, bit1.size()){
		printf("%d %lld\n", bit1[i].fi, bit1[i].se);
	}
	rep(i, bit2.size()){
		printf("%d %lld\n", bit2[i].fi, bit2[i].se);
	}
	printf("----\n");
	*/
	vector<pair<ll, ll> > ans;
	for (int i = 0; i < bit1.size(); ++i){//bit1のペアをbit2から探す
		auto itr = lower_bound(all(bit2), mp(-1, s - bit1[i].se));
		for (auto it = itr; it < bit2.end(); ++it){
			if(it->se + bit1[i].se == s){
				ans.pb(mp(bit1[i].fi, it->fi));
				// printf("%d %d\n", bit1[i].fi, it->fi);
			}else if(it->se + bit1[i].se > s){
				break;
			}
		}
	}

	vector<vector<int> > memo;
	for(auto u : ans){
		int mask1 = u.fi, mask2 = u.se;
		vector<int> tmp;
		for (int pos = 0; pos < mid1; ++pos){
			if(mask1 & (1 << pos)){
				tmp.pb(1);
			}
		}
		for (int pos = 0; pos < mid2; ++pos){
			if(mask2 & (1 << pos)){
				tmp.pb(1);
			}
		}
		memo.pb(tmp);
	}
	for(auto u : memo){
		for (int i = 0; i < n; ++i){
			if(u[i]){
				printf("%d ", i + 1);
			}
		}
		printf("\n");
	}
	return 0;
}
0