Submission #847587

#	Time	Username	Problem	Language	Result	Execution time	Memory
847587		Ahmed_Salah7	Bank (IZhO14_bank)	C++17	100 / 100	337 ms	4132 KiB

bank

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

const int N = 2e5 + 5, M = 1e6 + 6, mod = 998244353;
vector<int> mask[22];
int a[22], b[22];
bool vis[(1 << 20)];

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    // freopen("bank.in","r",stdin);
    //freopen("bank.out","w",stdout);
    int n, m;
    cin >> n >> m;
    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 < n; ++i) {
        for (int j = 0; j < (1 << m); ++j) {
            int sum = 0;
            for (int k = 0; k < m; ++k) {
                sum += b[k] * ((j >> k) & 1);
            }
            if (sum == a[i])
                mask[i].push_back(j);
        }
    }
    vector<int> masks;
    masks.push_back(0);
    for (int i = 0; i < n; ++i) {
        vector<int> _masks;
        for (auto x: mask[i]) {
            for (auto y: masks) {
                if (!(x & y) && !vis[x | y]) {
                    _masks.push_back(x | y);
                    vis[x | y] = 1;
                }
            }
        }
        masks = _masks;
    }
    cout << (masks.empty() ? "NO" : "YES");
}

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