Submission #1094933

#	Time	Username	Problem	Language	Result	Execution time	Memory
1094933		rayan_bd	Bank (IZhO14_bank)	C++17	71 / 100	1057 ms	201228 KiB

bank

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;

typedef tree<int, null_type, less<int>, rb_tree_tag, tree_order_statistics_node_update> pbds; 


#define getar(ar,n) for(ll i=0;i<n;++i) cin>>ar[i]
#define show(n) cout<<n<<'\n'
#define all(v) v.begin(), v.end()
#define br cout<<"\n"
#define pb push_back
#define nl '\n'
#define yes cout<<"YES\n"
#define no cout<<"NO\n"
#define ret return
#define ll long long
#define ld long double
#define sza(x) ((int)x.size())

const int mxN = 1048576;
const ll MOD = 1e9 + 7;
const ll INF = 1e9;
const ld EPS = 1e-9;

ll need[20],curr[20];
vector<ll> money[mxN];
ll dp[20][mxN];

bool f(ll i,ll mask){
	if(i<0) return 1;
	if(money[need[i]].size()==0) return 0;
	bool ok=0;
	for(auto it:money[need[i]]){
		bool good=1;
		for(ll j=0;j<31;++j){
			if((mask&(1ll<<j))&&(it&(1ll<<j))){
				good=0;
				break;
			}
		}
		if(good){
			ok|=f(i-1,mask|it);
		}
	}
	return ok;
}

void solve(ll tc) {
  ll n,m;cin>>n>>m;
  memset(dp,-1,sizeof(dp));

  for(ll i=0;i<n;++i) cin>>need[i];
  for(ll i=0;i<m;++i) cin>>curr[i];
  
  for(ll mask=0;mask<=(1ll<<m);++mask){
  	ll tot=0;
  	for(ll i=0;i<m;++i){
  		if(mask&(1ll<<i)) tot+=curr[i];
  	}
  	money[tot].pb(mask);
  }
  if(f(n-1,0)) yes;
  else no;

}

int main() {

    ios_base::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
    solve(0);
    return 0;
}

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