Submission #550221

#	Time	Username	Problem	Language	Result	Execution time	Memory
550221		Jomnoi	Bank (IZhO14_bank)	C++17	19 / 100	102 ms	8512 KiB

bank

#include <bits/stdc++.h>
#define DEBUG false
using namespace std;

const int MAX_N = 20;

int a[MAX_N], b[MAX_N];
pair <int, int> dp[1<<MAX_N];

int main() {
    cin.tie(nullptr)->sync_with_stdio(false);
    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 < (1<<M); i++) {
        dp[i] = make_pair(-1, -1);
    }

    dp[0] = make_pair(0, 0);
    for(int mask = 0; mask < (1<<M); mask++) {
        for(int i = 0; i < M; i++) {
            int prev = mask ^ (1<<i);
            if(dp[prev].first == -1) {
                continue;
            }

            auto [last, sum] = dp[prev];
            if(sum + b[i] == a[last]) {
                dp[mask] = make_pair(last + 1, 0);
            }
            else {
                dp[mask] = make_pair(last, sum + b[i]);
            }
        }

        if(dp[mask].first == N) {
            cout << "YES";
            return 0;
        }
    }
    cout << "NO";
    return 0;
}

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