Submission #1142413

#	Time	Username	Problem	Language	Result	Execution time	Memory
1142413		hoa208	Bank (IZhO14_bank)	C++20	100 / 100	93 ms	16712 KiB

bank

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define FOR(i, a, b) for (int i = (a), _b = (b); i <= _b; i++)
#define FORD(i, b, a) for (int i = (b), _a = (a); i >= _a; i--)
#define pa pair<ll, ll>
#define fi first
#define se second
#define bit(mask, j) ((mask >> j) & 1)
#define t_test int t;cin >> t;while(t--)
const ll mod = 1e9 + 7;
const ll INF = 1e17;
const ll N = 20;
ll n, m;
ll a[N + 5], b[N + 5];
namespace subX {
    const ll M = 1005;
    ll conver[1 << N], remain[1 << N];
    void slove() {
        memset(conver, -1, sizeof conver);
        memset(remain, -1, sizeof remain);
        conver[0] = remain[0] = 0;
        FOR(mask, 0, (1 << m) - 1) {
            ll id = conver[mask] + 1;
            if(conver[mask] == -1) continue;
            FOR(j, 0, m - 1) {
                if(bit(mask, j)) continue;
                if(remain[mask] + b[j + 1] < a[id]) {
                    conver[mask ^ (1 << j)] = conver[mask];
                    remain[mask ^ (1 << j)] = remain[mask] + b[j + 1]; 
                }
                else if(remain[mask] + b[j + 1] == a[id]) {
                    conver[mask ^ (1 << j)] = conver[mask] + 1;
                    remain[mask ^ (1 << j)] = 0;
                }
            }
            if(conver[mask] == n) {
                cout << "YES";
                return;
            }
            // cout << mask << ' ' << conver[mask] << ' ' << remain[mask] << '\n';
        }
        cout << "NO";
    }
}

void hbmt() {
    // inp
    cin >> n >> m;
    FOR(i, 1, n) {
        cin >> a[i];
    }
    FOR(i, 1, m) cin >> b[i];
    // slove
    return subX::slove();
}

int main() {
    
    ios_base::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);

    if(fopen("hbmt.inp", "r")) {
        freopen("hbmt.inp", "r", stdin);
        freopen("hbmt.out", "w", stdout);
    }
    
    // t_test 
    hbmt();
    return 0;
}

Compilation message (stderr)

bank.cpp: In function 'int main()':
bank.cpp:64:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   64 |         freopen("hbmt.inp", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
bank.cpp:65:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   65 |         freopen("hbmt.out", "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~

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