Submission #1325528

#TimeUsernameProblemLanguageResultExecution timeMemory
1325528brendonwBank (IZhO14_bank)C++20
0 / 100
1095 ms680 KiB
#include <bits/stdc++.h>

using namespace std;
int main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int n, m;
	cin >> n >> m;
	vector<int> a(n);
	for (auto &a_i: a) {
		cin >> a_i;
	}
	vector<int> b(m);
	for (auto &b_i: b) {
		cin >> b_i;
	}
	vector<vector<int>> masks(n);
	for (int mask = 0; mask < 1 << m; ++mask) {
		int sum = 0;
		for (int i = 0; i < m; ++i) {
			if ((mask) >> i & 1) {
				sum += b[i];
			}
		}
		for (int i = 0; i < n; ++i) {
			if (sum == a[i]) {
				masks[i].push_back(mask);
			}
		}
	}
	vector<vector<bool>> dp(n + 1, vector<bool>(1 << m));
	dp[0][0] = true;
	for (int i = 0; i < n; ++i) {
		for (int mask = 0; mask < 1 << m; ++mask) {
			for (auto &use: masks[i]) {
				if ((mask & use) == use) {
					if (dp[i][mask ^ use]) {
						dp[i + 1][mask] = true;
					}
					break;
				}
			}
		}
	}
	bool ans = false;
	for (int i = 0; i < 1 << m; ++i) {
		ans = ans || dp[n][i];
	}
	cout << (ans ? "YES" : "NO");

	return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...