結果

問題 No.15 カタログショッピング
ユーザー masamasa
提出日時 2015-01-22 15:48:22
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 87 ms / 5,000 ms
コード長 1,900 bytes
コンパイル時間 1,006 ms
コンパイル使用メモリ 86,488 KB
実行使用メモリ 14,300 KB
最終ジャッジ日時 2023-09-05 03:35:58
合計ジャッジ時間 1,936 ms
ジャッジサーバーID
(参考情報)
judge15 / judge14
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,380 KB
testcase_01 AC 2 ms
4,376 KB
testcase_02 AC 3 ms
4,376 KB
testcase_03 AC 14 ms
5,524 KB
testcase_04 AC 1 ms
4,380 KB
testcase_05 AC 80 ms
14,256 KB
testcase_06 AC 87 ms
14,212 KB
testcase_07 AC 86 ms
14,300 KB
testcase_08 AC 85 ms
14,116 KB
testcase_09 AC 84 ms
14,116 KB
権限があれば一括ダウンロードができます

ソースコード

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 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