Submission #1299847

#TimeUsernameProblemLanguageResultExecution timeMemory
1299847goppamachBank (IZhO14_bank)C++20
46 / 100
72 ms6408 KiB
#pragma GCC optimize("Ofast")

#include <bits/stdc++.h>

using namespace std;

#define el "\n"
#define pii pair<int, int>
#define pll pair<ll, ll>
#define fi first
#define se second
#define mp make_pair
#define sqr(x) ((x) * (x))
#define FOR(i, l, r) for (int i = l; i <= (r); i++)
#define FOD(i, l, r) for (int i = l; i >= (r); i--)
#define all(x) begin(x), end(x)
#define rall(x) rbegin(x), rend(x)
#define sz(x) ((int)(x).size())
#define fast_io ios_base::sync_with_stdio(false); cin.tie(nullptr);

using db = long double;
using ll = long long;
using ull = unsigned long long;
using vi = vector<int>;
using vll = vector<ll>;
using vpii = vector<pii>;
using vpll = vector<pll>;
using vvi = vector<vi>;
using vvll = vector<vll>;
using vbool = vector<bool>;
using vvbool = vector<vbool>;

template<class T> inline bool chmax(T &a, T const &b) { return (a < b ? (a = b, true) : false); }
template<class T> inline bool chmin(T &a, T const &b) { return (a > b ? (a = b, true) : false); }

// #define DEBUG
#ifdef DEBUG
#include "D:\cpp\debug.h"
#else
#define debug(...)
#define debug_arr(...)
#endif

mt19937 rng((uint32_t)chrono::steady_clock::now().time_since_epoch().count());

constexpr int N = 20;
constexpr int INF = 1E9 + 7;
constexpr ll INFLL = 4E18;
constexpr int MOD = 1E9 + 7; // 998244353
constexpr double EPS = 1E-10;

#define bit(i, mask) ((mask >> i) & 1)

bool dp[N + 1][1 << N];
int sum[1 << N];
int a[N + 1], b[N];
int n, m;

void solve() {
    cin >> n >> m;

    int salary_sum = 0;
    FOR(i, 1, n) {
        cin >> a[i];
        salary_sum += a[i];
    }

    int banknote_sum = 0;
    FOR(i, 0, m - 1) {
        cin >> b[i];
        banknote_sum += b[i];
    }

    if (salary_sum > banknote_sum) {
        cout << "NO" << el;
        return;
    }

    const int full_mask = (1 << m) - 1;

    FOR(mask, 0, full_mask) {
        dp[0][mask] = true;
        FOR(i, 0, m - 1) {
            if (bit(i, mask)) {
                sum[mask] += b[i];
            }
        }
    }

    FOR(i, 1, n) {
        FOR(mask, 0, full_mask) {
            if (dp[i][mask]) continue;
            if (sum[mask] == a[i] && dp[i - 1][full_mask & ~mask]) {
                for (int overmask = mask; overmask <= full_mask; overmask = (overmask + 1) | mask) {
                    dp[i][overmask] = true;
                }
            }
        }
    }

    cout << (dp[n][full_mask] ? "YES" : "NO") << el;
}

int main() {
    fast_io
    #define LOCAL
    #ifndef LOCAL
    #define PROBLEM "bank"
    freopen(PROBLEM ".in", "r", stdin);
    freopen(PROBLEM ".out", "w", stdout);
    #endif

    int t = 1;
    // cin >> t;
    while (t--) solve();
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...