Submission #660076

#	Time	Username	Problem	Language	Result	Execution time	Memory
660076		bojackduy	은행 (IZhO14_bank)	C++14	71 / 100	854 ms	262144 KiB

bank

#include <iostream>
#include <queue>
#include <stack>
#include <algorithm>
#include <string.h>
#include <functional>
#define task ""
#define size() size() * 1ll 
#define all(x) (x).begin(), (x).end()
#define pb push_back
#define pii pair<int, int>
#define fi first
#define se second
#define MASK(x) (1LL << (x))
#define BIT(x,i) (((x) >> (i)) & 1)
#define numbit(x) __builtin_popcountll(x)

using namespace std;

typedef vector<int> vi;
typedef vector<pii> vii;
typedef long long ll;

template<class t>
bool mini(t &x,t y) {
    if (y < x) {
       x = y;
       return 1;
    }
return 0;
}
template<class t>
bool maxi(t &x,t y) {
    if (x < y) {
       x = y;
       return 1;
    }
return 0;
}

const int N = 1e6 + 1;
const int M = 1e3 + 1;
const long long mod = 1e9 + 7;
const long long oo = 1e18 + 7;

int n, m;
int a[20], b[20];
vi v[N], d[2];
void solve(int test = -1) {
    cin >> n >> m;
    for (int i = 0; i < n; i++) cin >> a[i];
    for (int i = 0; i < m; i++) cin >> b[i];
    sort(a, a + n), sort(b, b + m);

    for (int i = 0; i < n; i++) {
        for (int mask = 0; mask < MASK(m); mask++) {
            int sum = 0;
            for (int msk = mask; msk > 0; msk -= msk & -msk) {
                int j = __builtin_ctz(msk);
                sum += b[j];
            }
            if (sum == a[i]) v[i].pb(mask);
        }
    }
    d[1] = v[0];
    for (int i = 1; i < n; i++) {
        if (!d[i & 1].size()) return void (cout << "NO");
        for (int mask: d[i & 1]) {
            for (int mask1: v[i]) {
                if (!(mask & mask1)) {
                    d[(i & 1) ^ 1].pb(mask | mask1);
                }
            }
        }
        d[i & 1] = vi(0);
    }
    if (!d[n & 1].size()) return void (cout << "NO");
    cout << "YES";
}

int32_t main()  {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    // freopen(task".inp", "r", stdin);
    // freopen(task".out", "w", stdout);

    int T = 1;
    // cin >> T;
    for (int i = 1; i <= T; i++) {
        solve(i);
    }
return 0;
}
/*

*/

Compilation message (stderr)

bank.cpp: In function 'void solve(int)':
bank.cpp:8:23: warning: '*' in boolean context, suggest '&&' instead [-Wint-in-bool-context]
    8 | #define size() size() * 1ll
      |                       ^
bank.cpp:67:23: note: in expansion of macro 'size'
   67 |         if (!d[i & 1].size()) return void (cout << "NO");
      |                       ^~~~
bank.cpp:8:23: warning: '*' in boolean context, suggest '&&' instead [-Wint-in-bool-context]
    8 | #define size() size() * 1ll
      |                       ^
bank.cpp:77:19: note: in expansion of macro 'size'
   77 |     if (!d[n & 1].size()) return void (cout << "NO");
      |                   ^~~~

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