結果

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

ソースコード

diff #
raw source code 

#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 it = maps[0].begin(); it != maps[0].end(); it++) {
		int rest = s - it->first;
		if (maps[1].count(rest)) {
			for (int i = 0; i < it->second.size(); i++) {
				for (int j = 0; j < maps[1][rest].size(); j++) {
					ans.push_back(it->second[i] | (maps[1][rest][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++) {
		bool first = true;
		for (int j = n_max - 1; j >= 0; j--) {
			if (ans[i] & (1 << j)) {
				if (!first) {
					printf(" ");
				}
				printf("%d", n_max - j);
				first = false;
			}
		}
		printf("\n");
	}
	return 0;
}
0