Submission #584392

#	Time	Username	Problem	Language	Result	Execution time	Memory
584392		stevancv	Bank (IZhO14_bank)	C++14	100 / 100	167 ms	8656 KiB

bank

#include <bits/stdc++.h>
#define ll long long
#define ld long double
#define sp ' '
#define en '\n'
#define smin(a, b) a = min(a, b)
#define smax(a, b) a = max(a, b)
using namespace std;
const int N = 1e5 + 2;
int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int n, m;
    cin >> n >> m;
    vector<int> a(n), b(m);
    for (int i = 0; i < n; i++) {
        cin >> a[i];
        if (i > 0) a[i] += a[i - 1];
    }
    for (int i = 0; i < m; i++) {
        cin >> b[i];
    }
    int M = (1 << m);
    vector<int> sum(M);
    for (int i = 0; i < M; i++) {
        for (int j = 0; j < m; j++) {
            if ((1 << j) & i) sum[i] += b[j];
        }
    }
    vector<int> dp(M, -1e9);
    dp[0] = 0;
    for (int i = 1; i < M; i++) {
        for (int j = 0; j < m; j++) {
            if ((1 << j) & i) smax(dp[i], dp[i ^ (1 << j)]);
        }
        for (int j = 0; j < n; j++) if (a[j] == sum[i]) dp[i]++;
    }
    for (int i = 0; i < M; i++) {
        if (dp[i] == n) {
            cout << "YES" << '\n';
            return 0;
        }
    }
    cout << "NO" << '\n';
    return 0;
}

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