Submission #171931

#	Time	Username	Problem	Language	Result	Execution time	Memory
171931		davitmarg	Bank (IZhO14_bank)	C++17	71 / 100	1074 ms	64380 KiB

bank

/*DavitMarg*/
#include <iostream>
#include <algorithm>
#include <cmath>
#include <vector>
#include <string>
#include <cstring>
#include <map>
#include <unordered_map>
#include <set>
#include <queue>
#include <iomanip>
#include <bitset>
#include <stack>
#include <cassert>
#include <iterator>
#include <fstream>
#define mod 1000000007ll
#define LL long long
#define LD long double
#define MP make_pair
#define PB push_back
#define all(v) v.begin(), v.end()
using namespace std;

const int N = 25;

int n, m, dp[N][(1 << 20) + 5];
int a[N], b[N], sum[(1 << 20) + 5];
vector<int> masks[N * 1003];

int main()
{
    cin >> n >> m;
    for (int i = 1; i <= n; i++)
        cin >> a[i];
    for (int i = 1; i <= m; i++)
        cin >> b[i];
    for (int mask = 0; mask < (1 << m); mask++)
    {
        for (int i = 0; i < m; i++)
            if (((1 << i) & mask))
                sum[mask] += b[i + 1];
        masks[sum[mask]].PB(mask);
        //cout << mask << " = " << sum[mask] << endl;
    }

    dp[0][0] = 1;
    for (int i = 1; i <= n; i++)
    {
        for (int j = 0; j < masks[a[i]].size(); j++)
        {
            int mask = masks[a[i]][j];
            int s = (1 << m) - 1 - mask;
            while (s >= 0)
            {
                if (dp[i - 1][s])
                    dp[i][mask | s] = 1;
                if (s == 0)
                    break;
                s = (s - 1) & ((1 << m) - 1 - mask);
            }
        }
        //for (int mask = 0; mask < (1 << m); mask++)
        //   cout << i << " : " << mask << " = " << dp[i][mask] << endl;
    }
    int ans = 0;
    for (int mask = 0; mask < (1 << m); mask++)
        ans += dp[n][mask];
    if (ans)
        cout << "YES" << endl;
    else
        cout << "NO" << endl;
    return 0;
}

/*
 
2 4
9 10
5 4 5 5
 
000 0
001 1
010 2
011 3
100 4
101 5
110 6
111 7
 
*/

Compilation message (stderr)

bank.cpp: In function 'int main()':
bank.cpp:51:27: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for (int j = 0; j < masks[a[i]].size(); j++)
                         ~~^~~~~~~~~~~~~~~~~~~~

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