Submission #1077883

#	Time	Username	Problem	Language	Result	Execution time	Memory
1077883		TsotneSV	Bank (IZhO14_bank)	C++17	100 / 100	324 ms	43436 KiB

bank

#pragma gcc diagnostic "-std=c++1z"
#include <bits/stdc++.h>
using namespace std;
/* /\_/\
  (= ._.)
  / >  \>
*/
//#pragma comment(linker, "/stack:200000000")
#pragma GCC optimize("Ofast")
#pragma GCC optimize ("unroll-loops")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native") // codeforces

// #define int long long
#define fi first
#define se second
#define pb push_back
#define ins insert
#define mp make_pair
#define send {ios_base::sync_with_stdio(false);}
#define help {cin.tie(0);}
#define endl '\n'
#define sz(x) ((long long) (x).size())
#define all(x) (x).begin(),(x).end()
#define print(x) cout<<(x)<<" ";
#define printl(x) cout<<(x)<<endl
#define dbg(x) cerr<<#x<<" "<<x<<endl

typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
typedef vector<int> vi;
typedef vector<ll> vll;
typedef vector<pii> vpi;
typedef vector<pll> vpl;

void fileIO(string filename) {
    freopen((filename + ".in").c_str(), "r", stdin);
    freopen((filename + ".out").c_str(), "w", stdout);
}

// const ll mod = 1000000007;
// const ll mod = 998244353;
// ll mod;

const int inf=1e9,MAXN=(1<<20) + 5; 
const ll INF=1e18; 
const ld pi = 3.14159265358979323846;

int n,m,A[22],B[22];
vi mask[20005]; bool vis[22][MAXN],dp[22][MAXN];

bool go(int i,int m) {

    if(i == n) return 1;

    if(vis[i][m]) return dp[i][m];

    vis[i][m] = 1;

    bool ret = 0;

    for(int b : mask[A[i]]) {

        if((m & b) == b) ret |= go(i+1,m^b);

    } return dp[i][m] = ret;

}

void solve(int tc = 0){
    cin>>n>>m; memset(vis,0,sizeof(vis));

    for(int i=0;i<n;i++) cin>>A[i];
    for(int i=0;i<m;i++) cin>>B[i];

    for(int i=0;i<(1<<m);i++) {
        int sum = 0;
        for(int j=0;j<m;j++) {
            if((1<<j)&i) sum += B[j];
        } mask[sum].pb(i);
    }

    printl(go(0,(1<<m)-1) ? "YES" : "NO");

}

signed main(){
  
    send help
  
    int tc=1;
    // cin>>tc;
    for(int t = 0; t < tc; t++) solve(t);
}

Compilation message (stderr)

bank.cpp:1: warning: ignoring '#pragma gcc diagnostic' [-Wunknown-pragmas]
    1 | #pragma gcc diagnostic "-std=c++1z"
      | 
bank.cpp: In function 'void fileIO(std::string)':
bank.cpp:39:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   39 |     freopen((filename + ".in").c_str(), "r", stdin);
      |     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
bank.cpp:40:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   40 |     freopen((filename + ".out").c_str(), "w", stdout);
      |     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

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