Submission #914436

#	Time	Username	Problem	Language	Result	Execution time	Memory
914436		thisistotallymyusername	Bank (IZhO14_bank)	C++14	0 / 100	105 ms	175248 KiB

bank

#include <bits/stdc++.h>

using namespace std;

const int N = 21, M = 1 << (N - 1), K = 1010;

int dp[N][M];
int n, m, mx, a[N], b[N];
bool st[K];
vector<int> s[K];

bool check(int i, int msk) {
    for (int j = 0; j < N; j++) {
        if (i >> j & 1 && !(msk >> j & 1)) {
            return false;
        }
    }
    return true;
}

int g(int i, int msk) {
    for (int j = 0; j < N; j++) {
        if (i >> j & 1) {
            msk = msk & ~(1 << j);
        }
    }
    return msk;
}

bool f(int u, int msk) {
    if (dp[u][msk] == 1 || u == 0) {
        return true;
    } if (dp[u][msk] == 0) {
        return false;
    }
    for (int i : s[a[u]]) {
        if (check(i, msk) && f(u - 1, g(i, msk))) {
            dp[u][msk] = 1;
            return true;
        }
    }
    dp[u][msk] = 0;
    return false;
}

int main() {
    cin.tie(0)->sync_with_stdio(0);
    memset(dp, -1, sizeof dp);
    cin >> n >> m;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
        st[a[i]] = true;
    }
    for (int j = 1; j <= m; j++) {
        cin >> b[j];
    }
    for (int i = 0; i < 1 << m; i++) {
        int sum = 0;
        for (int j = 0; j < N; j++) {
            if (i >> j & 1) {
                sum += b[j + 1];
            }
        }
        if (st[sum]) {
            s[sum].emplace_back(i);
        }
    }
    cout << (f(n, (1 << m) - 1) ? "YES\n" : "NO\n");
    return 0;
}

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