Submission #954099

#	Time	Username	Problem	Language	Result	Execution time	Memory
954099		Ariadna	Bank (IZhO14_bank)	C++14	25 / 100	1082 ms	262144 KiB

bank

#include <bits/stdc++.h>
#define mp make_pair

using namespace std;

bool pay(int n, int m, vector<int>& a, vector<int>& b) {
    if (m < n) return false;

    vector<bool> dp(n);
    int mask = 0;
    queue<pair<int, pair<int, int>>> q;
    q.push(mp(0, mp(0, a[0])));
    while (!q.empty()) {
        int mask = q.front().first;
        int index = q.front().second.first;
        int sum = q.front().second.second;
        q.pop();

        for (int i = 0; i < m; ++i) {
            if (!(mask & (1<<i))) {
                if (b[i] < sum) {
                    q.push(mp(mask | (1<<i), mp(index, sum - b[i])));
                } else if (b[i] == sum) {
                    dp[index] = true;
                    if (index+1 < n) q.push(mp(mask | (1<<i), mp(index+1, a[index+1])));
                }
            }
        }
    }
    return dp[n-1];
}

int main() {
    int n, m;
    cin >> n >> m;
    vector<int> a(n), b(m);
    for (int i = 0; i < n; ++i) cin >> a[i];
    for (int j = 0; j < m; ++j) cin >> b[j];
    if (pay(n, m, a, b)) cout << "YES\n";
    else cout << "NO\n";

    return 0;
}

Compilation message (stderr)

bank.cpp: In function 'bool pay(int, int, std::vector<int>&, std::vector<int>&)':
bank.cpp:10:9: warning: unused variable 'mask' [-Wunused-variable]
   10 |     int mask = 0;
      |         ^~~~

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