Submission #421222

#	Time	Username	Problem	Language	Result	Execution time	Memory
421222		antimirage	Bank (IZhO14_bank)	C++14	100 / 100	155 ms	4488 KiB

bank

#include <bits/stdc++.h>

#define fr first
#define sc second
#define mk make_pair
#define pb push_back

using namespace std;

int n, m, dp[(1 << 20) + 5], a[25], b[25];

int main (){
    cin >> n >> m;
    for (int i = 1; i <= n; i++) {
        scanf("%d", &a[i]);
        a[i] += a[i - 1];
    }
    for (int i = 0; i < m; i++) {
        scanf("%d", &b[i]);
    }
    memset(dp, -1, sizeof(dp));
    dp[0] = 0;
    for (int mask = 0; mask < (1 << m); mask++) {
        if (dp[mask] == -1) continue;
        if (dp[mask] == n) {
            puts("YES");
            return 0;
        }
        int sum = 0;
        for (int j = 0; j < m; j++) {
            if (mask & (1 << j))
                sum += b[j];
        }
        for (int j = 0; j < m; j++) {
            if (mask & (1 << j)) continue;
            
            if (sum + b[j] < a[dp[mask] + 1]) {
                dp[mask | (1 << j)] = dp[mask];
            } else if (sum + b[j] == a[dp[mask] + 1]) {
                dp[mask | (1 << j)] = dp[mask] + 1;
            }
        }
    }
    puts("NO");
}

Compilation message (stderr)

bank.cpp: In function 'int main()':
bank.cpp:15:14: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   15 |         scanf("%d", &a[i]);
      |         ~~~~~^~~~~~~~~~~~~
bank.cpp:19:14: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   19 |         scanf("%d", &b[i]);
      |         ~~~~~^~~~~~~~~~~~~

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