제출 #1358269

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1358269		GGOSHAB	은행 (IZhO14_bank)	C++20	100 / 100	80 ms	16716 KiB

bank

#ifdef ONPC
    #define _GLIBCXX_DEBUG
#endif // ONPC
#include <bits/stdc++.h>

using namespace std;

#define all(v) begin(v), end(v)

using ll = long long;

const int inf = (int)1e9 + 10;
const ll INF = (ll)1e18 + 10;

const int mod = 1e9 + 7;

void test_case() {
    int n, m;
    cin >> n >> m;
    vector<ll> a(n), b(m);
    for (auto &i : a) {
        cin >> i;
    }
    for (auto &i : b) {
        cin >> i;
    }

    vector<pair<ll, ll>> dp(1 << m);
    for (int mask = 0; mask < (1 << m); ++mask) {
        for (int i = 0; i < m; ++i) {
            if (mask >> i & 1) {
                auto [pref, sum] = dp[mask ^ (1 << i)];
                if (pref == n) {
                    dp[mask] = max(dp[mask], make_pair(pref, sum + b[i]));
                } else {
                    if (sum + b[i] == a[pref]) {
                        dp[mask] = max(dp[mask], make_pair(pref + 1, sum + b[i] - a[pref]));
                    } else {
                        dp[mask] = max(dp[mask], make_pair(pref, sum + b[i]));
                    }
                }
            }
        }
    }

    cout << (dp.back().first == n ? "YES" : "NO") << '\n';
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);

    int t = 1;
    // cin >> t;
    while (t--) {
        test_case();
    }

    return 0;
}

• 서브태스크 1: 19 / 19
• 서브태스크 2: 25 / 25
• 서브태스크 3: 27 / 27
• 서브태스크 4: 29 / 29