제출 #1312331

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1312331		tkhoi13	은행 (IZhO14_bank)	C++20	44 / 100	6 ms	7216 KiB

bank

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#define ll long long
#define db double
#define pii pair<int, int>
#define pll pair<ll, ll>
#define fi first
#define se second
#define pb push_back
#define all(x) begin(x), end(x)
#define allr(x) rbegin(x), rend(x)
#define szx(x) ((int)(x).size())
#define FOR(i, a, b) for (int i = a, _b = (b); i <= _b; ++i)
#define ROF(i, a, b) for (int i = a, _b = (b); i >= _b; --i)
#define REP(i, n) for (int i = 0, _n = (n); i < _n; ++i)
#define endl '\n'
#define inf 1000000007
#define mod 1000000007
// #define mod 998244353

using namespace std;
using namespace __gnu_pbds;

void setIO() {
    ios::sync_with_stdio(0);
    cin.tie(0);
}
void openFile(string filename = "") {
    if (!filename.empty())
        if (ifstream(filename + ".in")) {
            freopen((filename + ".in").c_str(), "r", stdin);
            freopen((filename + ".out").c_str(), "w", stdout);
        }
}
ll binpow(ll a, ll b, int m = mod) {
    ll res = 1;
    a %= m;
    while (b) {
        if (b & 1) res = (a * res) % m;
        a = (a * a) % m;
        b >>= 1;
    }
    return res;
}
const int FMAX = 2e5 + 5;
ll fact[FMAX], invf[FMAX];
void process() {
    fact[1] = fact[0] = invf[1] = invf[0] = 1;
    FOR(i, 2, FMAX - 1) fact[i] = (fact[i - 1] * i) % mod;
    invf[FMAX - 1] = binpow(fact[FMAX - 1], mod - 2, mod);
    ROF(i, FMAX - 1, 1) invf[i - 1] = invf[i] * i % mod;
}
ll comb(int a, int b) {
    b = min(b, a - b);
    if (a < 0 || b < 0) return 0;
    return fact[a] * (invf[b] * invf[a - b] % mod) % mod;
}
const int PMAX = 1e6 + 5;
bool prime[PMAX + 1];
void sieve() {
    fill(prime, prime + PMAX + 1, 1);
    prime[0] = prime[1] = 0;
    for (int i = 2; i * i <= PMAX; i++) {
        if (prime[i] == 0) continue;
        prime[i] = 1;
        for (ll j = i * i; j <= PMAX; j += i) prime[j] = 0;
    }
}
struct DSU {
    vector<int> comp;
    ll ans = 0;
    DSU(int n) { comp.assign(n + 5, -1); }

    int find(int v) { return comp[v] < 0 ? v : comp[v] = find(comp[v]); }

    void unite(int u, int v) {
        u = find(u), v = find(v);
        if (u == v) return;

        if (comp[u] > comp[v]) swap(u, v);
        comp[u] += comp[v];
        comp[v] = u;
    }

    int getsz(int v) { return -comp[find(v)]; }
    bool same(int u, int v) { return find(u) == find(v); }
};

const int MAX = 21;
int n, m;
int a[MAX], b[MAX];
vector<int> construct[MAX];
int sum[1 << MAX];
bool dp[1 << MAX][2];

void solve() {
    REP(mask, 1 << m) {
        int i = __builtin_ctz(mask);
        sum[mask] += sum[mask ^ (1 << i)] + b[i];
    }

    REP(i, n) REP(mask, 1 << m) if (sum[mask] == a[i]) construct[i].pb(mask);

    dp[0][0] = 1;

    bool last;

    REP(i, n) {
        bool cur = i & 1;
        bool next = !cur;

        REP(mask, 1 << m) {
            if (!dp[mask][cur]) continue;
            for (int j : construct[i])
                if ((mask & j) == 0) dp[mask | j][next] = 1;
        }

        REP(mask, 1 << m) dp[mask][cur] = 0;
        if (i == n - 1) last = next;
    }

    bool ok = 0;

    REP(i, 1 << m) ok |= dp[i][last];

    cout << (ok ? "YES" : "NO");
}

void input() {
    cin >> n >> m;
    REP(i, n) cin >> a[i];
    REP(i, m) cin >> b[i];
}

void preprocess() { input(); }

int main() {
    setIO();
    openFile("main");
    int t = 1;
    // cin >> t;
    preprocess();
    while (t--) { solve(); }
}

컴파일 시 표준 에러 (stderr) 메시지

bank.cpp: In function 'void openFile(std::string)':
bank.cpp:32:20: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   32 |             freopen((filename + ".in").c_str(), "r", stdin);
      |             ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
bank.cpp:33:20: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   33 |             freopen((filename + ".out").c_str(), "w", stdout);
      |             ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

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