Submission #919679

#	Time	Username	Problem	Language	Result	Execution time	Memory
919679		dubabuba	Bank (IZhO14_bank)	C++14	100 / 100	97 ms	186704 KiB

bank

#include <bits/stdc++.h>
using namespace std;
 
const int mxn = 21;
const int mxa = 1010;
vector<int> bit[mxa * mxn];
int sum[1 << mxn];
int n, a[mxn];
int m, b[mxn];
 
int dp[mxn][1 << mxn];
int solve(int i, int cur) {
	if(i == 0) return 1;
	if(dp[i][cur] > -1) return dp[i][cur];
 
	int &val = a[i];
	dp[i][cur] = 0;
	for(int las : bit[val]) {
		if(((cur | las) == cur) && solve(i - 1, cur ^ las))
			dp[i][cur] = 1;
	}
	return dp[i][cur];
}
 
int main() {
	cin >> n >> m;
	for(int i = 1; i <= n; i++) cin >> a[i];
	for(int j = 0; j < m; j++) cin >> b[j];
 
	for(int cur = 0; cur < (1 << m); cur++) {
		int sus, prv;
		for(sus = 0; sus < m; sus++)
			if(cur & (1 << sus)) break;
		prv = cur ^ (1 << sus);
		sum[cur] = sum[prv] + b[sus];
		bit[sum[cur]].push_back(cur);
	}
 	memset(dp, -1, sizeof dp);
	if(solve(n, (1 << m) - 1)) cout << "YES\n";
	else cout << "NO\n";
	return 0;
}

• Subtask 1: 19 / 19
• Subtask 2: 25 / 25
• Subtask 3: 27 / 27
• Subtask 4: 29 / 29