Submission #987853

#	Time	Username	Problem	Language	Result	Execution time	Memory
987853		vjudge1	Bank (IZhO14_bank)	C++17	19 / 100	137 ms	20004 KiB

bank

#include<bits/stdc++.h>

using namespace std;

#define ll long long
#define endl "\n" 
int n, m, a[25], b[25];
int dp[5000000];
int t[25];
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cin >> n >> m;
    memset(dp, -1, sizeof dp);
    int s = 0;
    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
        t[i] = t[i - 1] + a[i];
    }
    for (int i = 0; i < m; i++)
    {
        cin >> b[i];
    }
    dp[0] = 0;
    for (int mask = 0; mask < (1 << m); mask++)
    {
        int sum = 0;
        for (int j = 0; j < m; j++)
        {
            if (mask & (1 << j))
            {
                sum += b[j];
            }
        }
        // cout << mask << " " << sum << endl;
        for (int j = 0 ; j < m; j++)
        {
            if (!(mask & (1 << j)))
            {
                int newmask = mask ^ (1 << j);
                // cout << "* " << newmask << " " << sum + b[j] << endl;
                if (sum + b[j] == t[dp[mask] + 1])
                {
                    dp[newmask] = dp[mask] + 1;
                }
                else dp[newmask] = dp[mask];
            }
        }
    }
    // cout << *max_element(dp, dp + (1 << m) + 1);
    if (*max_element(dp, dp + (1 << m) + 1) == n) cout << "YES";
    else cout << "NO";
}

Compilation message (stderr)

bank.cpp: In function 'int main()':
bank.cpp:15:9: warning: unused variable 's' [-Wunused-variable]
   15 |     int s = 0;
      |         ^

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