#230131 (C++14) No.617 Nafmo、買い出しに行く

• 問題: No.617 Nafmo、買い出しに行く
• ユーザー: masa
• 提出日時: 2018-01-17 18:36:54
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 4 ms / 2,000 ms
• コード長: 860 bytes
• コンパイル時間: 679 ms
• コンパイル使用メモリ: 81,936 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-06-06 14:37:52
• 合計ジャッジ時間: 1,418 ms
• ジャッジサーバーID （参考情報）: judge5 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	1 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	AC	2 ms; 5,376 KB
testcase_07	AC	1 ms; 5,376 KB
testcase_08	AC	1 ms; 5,376 KB
testcase_09	AC	2 ms; 5,376 KB
testcase_10	AC	3 ms; 5,376 KB
testcase_11	AC	3 ms; 5,376 KB
testcase_12	AC	3 ms; 5,376 KB
testcase_13	AC	3 ms; 5,376 KB
testcase_14	AC	4 ms; 5,376 KB
testcase_15	AC	2 ms; 5,376 KB
testcase_16	AC	2 ms; 5,376 KB
testcase_17	AC	2 ms; 5,376 KB
testcase_18	AC	2 ms; 5,376 KB
testcase_19	AC	1 ms; 5,376 KB
testcase_20	AC	2 ms; 5,376 KB
testcase_21	AC	2 ms; 5,376 KB

ソースコード

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
#include <utility>
#include <string>
#include <set>

using namespace std;

vector<int> calc_weights(vector<int> a, int k) {
	set<int> ws;

	int n = a.size();
	int limit = 1 << n;
	for (int i = 0; i < limit; i++) {
		int w = 0;
		for (int j = 0; j < n; j++) {
			if ((i >> j) & 1) {
				w += a[j];
			}
		}
		if (w <= k) {
			ws.insert(w);
		}
	}

	return vector<int>(ws.begin(), ws.end());
}
int main() {
	int n, k, a;
	cin >> n >> k;

	int n2 = n / 2;
	vector<int> as1, as2;
	for (int i = 0; i < n; i++) {
		cin >> a;
		(i < n2 ? as1 : as2).emplace_back(a);
	}

	int ans = 0;
	auto w1 = calc_weights(as1, k);
	auto w2 = calc_weights(as2, k);

	for (auto x : w1) {
		for (auto y : w2) {
			if (x + y <= k) {
				ans = max(ans, x + y);
			}
		}
	}

	cout << ans << endl;
	return 0;
}