Submission #1218267

#	Time	Username	Problem	Language	Result	Execution time	Memory
1218267		goulthen	Bank (IZhO14_bank)	C++20	100 / 100	106 ms	17044 KiB

bank

//#pragma GCC optimize("O3")
//#pragma GCC optimize("unroll-loops")
#include <bits/stdc++.h>
using namespace std;
#include <ext/pb_ds/assoc_container.hpp>
using namespace __gnu_pbds;
template <class T>
using Tree =
    tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;

#define int long long
#define ll long long
#define pii pair<int, int>
#define pll pair<ll, ll>
#define fi first
#define se second
#define rep(i, a, b) for (int i = a; i <= b; ++i)
#define per(i, b, a) for (int i = b; i >= a; --i)
#define pb push_back
#define eb emplace_back
#define all(v) (v).begin(), (v).end()
#define lsb(x) (x)&(-x)
 
void setIO(string name = "") {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);cout.tie(nullptr);
    if (!name.empty()) {
        freopen((name + ".in").c_str(), "r", stdin);
        freopen((name + ".out").c_str(), "w", stdout);
    }
}
ll fexp(ll a, ll b, ll m) {
    if (b == 0) return 1LL;
    ll p = a;
    ll ans = 1;
 
    while (b > 0) {
        if (b % 2 != 0) ans = (ans*p)%m;
        p = (p*p)%m;
        b >>= 1;
    }
    return ans;
}
 
const int MAXN = 21;
const int INF = 1e18+5;
const int MOD =  1e9+7;
pii dp[(1<<MAXN)];
int a[MAXN], v[MAXN];

void solve() {
    int n,m;cin >> n >> m;
    rep(i,1,n) cin >> a[i];
    rep(i,1,m) cin >> v[i];
    a[n+1]=INF;

    rep(mk,1,(1<<m)-1) {
        rep(j,0,m-1) {
            if(((1<<j)&mk)==0) continue;

            int k = dp[mk^(1<<j)].fi+1;
            int s = dp[mk^(1<<j)].se+v[j+1];
            dp[mk] = max(dp[mk], {k-1,s});
            if(s==a[k]) dp[mk] = max(dp[mk], {k,s-a[k]});
        }
    }

    cout << (dp[(1<<m)-1].fi==n?"YES\n":"NO\n");
}



int32_t main() {
    setIO();
    int tt = 1;
    //cin >> tt;

    while (tt-- > 0) solve();
    
}

Compilation message (stderr)

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

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