Submission #922257

#	Time	Username	Problem	Language	Result	Execution time	Memory
922257		Baizho	Bank (IZhO14_bank)	C++17	100 / 100	317 ms	15652 KiB

bank

#include <bits/stdc++.h>
using namespace std;
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
  
#define ordered_set tree<int, null_type,less<int>, rb_tree_tag,tree_order_statistics_node_update>
 
 #pragma GCC optimize("Ofast,unroll-loops,fast-math")
 #pragma GCC target("popcnt")
 
 
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
typedef pair<ll,ll> pll;
 
#define sz size()
#define ff first
#define ss second
#define all(a) a.begin(),a.end()
#define pb push_back
 
//const int mod = ll(1e9) + 7; //(b + (a%b)) % b (to mod -1%(10^9+7) correctly in c++ its -1 but its suppose to be 10^9+6
const int mod = 998244353;  // (a%mod)*(binpow(b,mod-2,mod) = (a/b)%mod
const int N = ll(2e5) + 11;
const long long inf = 1e9;
const long double eps = 1e-15L;
 
ll lcm(ll a, ll b) { return (a / __gcd(a,b))*b; }
 
ll binpow(ll a, ll b, ll m) { ll res=1; a %= m; while(b>0){ if(b&1)res=(res * a) % m; a=(a * a) % m; b/=2; } return res%m;}
 
int n, m, a[23], b[23];
bool dp[23][(1 << 20) + 10];
vector<int> ways[30];
 
void Baizho() {
	cin>>n>>m;
	for(int i = 1; i <= n; i ++) cin>>a[i];
	for(int i = 1; i <= m; i ++) cin>>b[i];
	for(int mask = 1; mask < (1 << m); mask ++) {
		int sum = 0;
		for(int i = 1; i <= m; i ++) if(mask & (1 << i - 1)) sum += b[i];
		for(int i = 1; i <= n; i ++) if(a[i] == sum) ways[i].pb(mask);
	}
	
	dp[0][0] = 1;
	for(int i = 0; i < n; i ++) {
		for(int mask = 0; mask < (1 << m); mask ++) {
			if(!dp[i][mask]) continue;
			for(auto way : ways[i + 1]) {
				if((mask & way) == 0) dp[i + 1][mask | way] |= dp[i][mask];
			}
		}
	}
	bool ok = 0;
	for(int i = 0; i < (1 << m); i ++) ok |= dp[n][i];
	if(ok) cout<<"YES\n";
	else cout<<"NO\n";
}
 
signed main() {		
// 	Freopen("hps");
    ios_base::sync_with_stdio(false);   
    cin.tie(0);cout.tie(0); 
//   	precalc();
   	
    int ttt = 1;
//    cin>>ttt;
 
    for(int i=1; i<=ttt; i++) {Baizho(); }
}

Compilation message (stderr)

bank.cpp: In function 'void Baizho()':
bank.cpp:44:50: warning: suggest parentheses around '-' inside '<<' [-Wparentheses]
   44 |   for(int i = 1; i <= m; i ++) if(mask & (1 << i - 1)) sum += b[i];
      |                                                ~~^~~

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