Submission #245346

#	Time	Username	Problem	Language	Result	Execution time	Memory
245346		kshitij_sodani	Bank (IZhO14_bank)	C++17	71 / 100	1090 ms	36884 KiB

bank

#include <bits/stdc++.h>
using namespace std;
typedef int64_t llo;
#define mp make_pair
#define pb push_back
#define a first 
#define b second
//#define endl '\n'
int n,m;
int dp[21][1<<20];
int aa[21];
int bb[21];
int su[1<<20];
vector<int> pre[1<<20];
int main(){
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cin>>n>>m;
	for(int i=0;i<n;i++){
		cin>>aa[i];
	}
	for(int i=0;i<m;i++){
		cin>>bb[i];
	}
	string ans="NO";
	for(int j=0;j<(1<<m);j++){
		for(int k=0;k<m;k++){
			if((1<<k)&j){
				su[j]=su[j-(1<<k)]+bb[k];
				break;
			}
		}
		for(int i=0;i<n;i++){
			if(aa[i]==su[j]){
				pre[i].pb(j);
				if(n==1){
					ans="YES";
				}
			}
		}
	}
	if(n==1){
		cout<<ans<<endl;
		return 0;
	}
	for(int j=0;j<(1<<m);j++){
		dp[n][j]=1;
	}
	for(int i=n-1;i>=0;i--){
		for(int j=0;j<(1<<m);j++){
			for(int k=j;k>0;k=(k-1)&j){
				if(su[k]==aa[i]){
					dp[i][j]|=dp[i+1][j-k];
					if(dp[i][j]==1){
						break;
					}
				}
			}
		}
	}
	if(dp[0][(1<<m)-1]){
		cout<<"YES"<<endl;
	}
	else{
		cout<<"NO"<<endl;
	}






	return 0;
}

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