Submission #1169976

#TimeUsernameProblemLanguageResultExecution timeMemory
1169976Truly_Bank (IZhO14_bank)C++20
71 / 100
1098 ms20548 KiB
#define nametask ""
#include <bits/stdc++.h>
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define SZ(a) (int) a.size()
#define all(a) a.begin(), a.end()
#define FOR(i, a, b) for (int i = a; i <= b; i++)
#define FOD(i, b, a) for (int i = b; i >= a; i--)

using namespace std;
typedef long long ll;
typedef pair <int, int> pi;
const int N = 20;

int n, m;
int a[N], b[N];
int sum[(1 << N)];
bool f[(1 << N)][21];

signed main()
{
    ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
    if (fopen(nametask".inp", "r")){
        freopen(nametask".inp", "r", stdin);
        freopen(nametask".out", "w", stdout);
    }
    cin >> n >> m;
    FOR(i, 0, n - 1) cin >> a[i];
    FOR(i, 0, m - 1) cin >> b[i];
    FOR(mask, 0, (1 << m) - 1){
        FOR(i, 0, m - 1){
            if ((mask >> i) & 1) sum[mask] += b[i];
        }
    }
    FOR(mask, 0, (1 << m) - 1){
        if (sum[mask] == a[0]) f[mask][0] = 1;
    }
    bool ck = 0;
    FOR(i, 1, n - 1){
        FOR(mask, 1, (1 << m) - 1){
            for (int smask1 = mask; smask1 > 0; smask1 = (smask1 - 1) & mask){
                int smask2 = mask ^ smask1;
                if (sum[smask1] == a[i]){
                    f[mask][i] |= f[smask2][i - 1];
                }
            }
        }
    }
    FOR(mask, 0, (1 << m) - 1) {
        if (f[mask][n - 1]) {
            cout << "YES\n";
            return 0;
        }
    }
    cout << "NO\n";
    return 0;
}

Compilation message (stderr)

bank.cpp: In function 'int main()':
bank.cpp:26:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   26 |         freopen(nametask".inp", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~
bank.cpp:27:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   27 |         freopen(nametask".out", "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...