Submission #671465

#	Time	Username	Problem	Language	Result	Execution time	Memory
671465		Alihan_8	Bank (IZhO14_bank)	C++17	100 / 100	709 ms	172792 KiB

bank

#include <bits/stdc++.h>
// include <ext/pb_ds/assoc_container.hpp>
// include <ext/pb_ds/tree_policy.hpp>
// using namespace __gnu_pbds;
using namespace std;
#define all(x) x.begin(), x.end()
#define pb push_back
// define ordered_set tree<int,null_type,less_equal<int>,rb_tree_tag,tree_order_statistics_node_update>
#define mpr make_pair
#define ln '\n'
void IO(string name){freopen((name+".in").c_str(),"r",stdin); freopen((name+".out").c_str(),"w",stdout);}
//#define int long long
signed main(){
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);

    int n, m; cin >> n >> m;
    vector <int> a(n), b(m);
    for ( auto &i: a ) cin >> i;
    for ( auto &i: b ) cin >> i;
    vector <int> g[n];
    for ( int mask = 1; mask < (1 << m); mask++ ){
        int cnt = 0;
        for ( int i = 0; i < m; i++ ){
            if ( mask >> i & 1 ) cnt += b[i];
        }
        for ( int j = 0; j < n; j++ ){
            if ( a[j] == cnt ) g[j].pb(mask);
        }
    }
    const int S = 1 << m;
    vector <vector<int>> dp(n, vector <int> (S)), used(n+1, vector <int> (S));
    function <int(int,int)> dfs = [&](int mask, int pos){
        if ( pos == n ) return 1;
        if ( used[pos][mask] ) return dp[pos][mask];
        used[pos][mask] = 1;
        int &res = dp[pos][mask]; res = 0;
        for ( auto val: g[pos] ){
            if ( (val & mask) ^ val ) continue;
            if ( dfs(mask^val, pos+1) ) res = 1;
        }
        return res;
    };
    dfs(S-1, 0);
    if ( *max_element(all(dp[n-1])) ) cout << "YES";
    else cout << "NO";

    cout << '\n';
}

Compilation message (stderr)

bank.cpp: In function 'void IO(std::string)':
bank.cpp:11:29: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   11 | void IO(string name){freopen((name+".in").c_str(),"r",stdin); freopen((name+".out").c_str(),"w",stdout);}
      |                      ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
bank.cpp:11:70: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   11 | void IO(string name){freopen((name+".in").c_str(),"r",stdin); freopen((name+".out").c_str(),"w",stdout);}
      |                                                               ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

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