結果

問題 No.15 カタログショッピング
コンテスト
ユーザー masa
提出日時 2015-01-22 15:26:04
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
TLE  
実行時間 -
コード長 1,878 bytes
コンパイル時間 1,085 ms
コンパイル使用メモリ 84,376 KB
実行使用メモリ 14,016 KB
最終ジャッジ日時 2024-06-23 00:19:21
合計ジャッジ時間 7,788 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 5 TLE * 1 -- * 4
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
#include <utility>
#include <map>
#include <bitset>

using namespace std;

const int n_max = 31;
const int n_mid = 16;

int main() {
	int n, s;

	cin >> n >> s;

	vector<int> p(n, 0);
	for (int i = 0; i < n; i++) {
		cin >> p[i];
	}

	vector< map<int, vector<int> > > maps(2); // (price, vector<bits>)のvector
	maps[0][0] = vector<int>(1,0);
	maps[1][0] = vector<int>(1,0);

	for (int idx = 0; idx < 2; idx++) {
		int n_bit, offset;
		if (idx == 0) {
			n_bit = min(n, n_mid);
			offset = 0;
		} else {
			n_bit = max(0, n - n_mid);
			offset = n_mid;
		}


		for (int i = 1; i < (1 << n_bit); i++) {
			int total = 0;
			for (int j = 0; j < n_bit; j++) {
				if (i & (1 << j)) {
					total += p[j + offset];
				}
			}
			if (total > s) {
				continue;
			}
			if (maps[idx].count(total)) {
				maps[idx][total].push_back(i);
			} else {
				maps[idx][total] = vector<int>(1, i);
			}
		}
	}

	vector<int> ans;
	for (auto it0 = maps[0].begin(); it0 != maps[0].end(); it0++) {
		for (auto it1 = maps[1].begin(); it1 != maps[1].end(); it1++) {
			if(it0->first + it1->first == s) {
				for (int i = 0; i < it0->second.size(); i++) {
					for (int j = 0; j < it1->second.size(); j++) {
						ans.push_back(it0->second[i] | (it1->second[j] << n_mid) );
					}
				}
			}
		}
	}

	for (int i = 0; i < ans.size(); i++) {
		int tmp = 0;
		// cout << static_cast< bitset<n_max> >(ans[i]) << endl;

		for (int j = 0; j < n_max; j++) {
			if (ans[i] & (1 << j)) {
				tmp |= 1 << (n_max - j - 1);
			}
		}
		ans[i] = tmp;
		// cout << static_cast< bitset<n_max> >(tmp) << endl;
	}

	sort(ans.begin(), ans.end(), greater<int>());
	for (int i = 0; i < ans.size(); i++) {
		for (int j = n_max - 1; j >= 0; j--) {
			if (ans[i] & (1 << j)) {
				printf("%2d ", n_max - j);
			}
		}
		printf("\n");
	}
	return 0;
}
0