제출 #978947

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
978947		KasymK	은행 (IZhO14_bank)	C++17	100 / 100	152 ms	8796 KiB

bank

#include "bits/stdc++.h"

using namespace std;

int main(){
    ios::sync_with_stdio(false); cin.tie(nullptr);
    int n, m;
    cin >> n >> m;
    vector<int> a(n + 1), pref(n + 1), b(m);
    for(int i = 1; i <= n; ++i)
        cin >> a[i], pref[i] = pref[i - 1] + a[i];
    for(int &i : b)
        cin >> i;
    vector<int> dp(1 << 21);
    dp[0] = 1;
    int ok = 0;
    for(int mk = 0; mk < (1 << m); ++mk){
        int sum = 0;
        for(int i = 0; i < m; ++i)
            sum += (mk >> i & 1 ? b[i] : 0);
        int id = lower_bound(pref.begin(), pref.end(), sum) - pref.begin();
        for(int i = 0; i < m; ++i)
            if((mk >> i & 1) and (b[i] <= sum - pref[id - 1]))
                dp[mk] |= dp[mk ^ (1 << i)];
        ok |= (pref[n] == sum and dp[mk]);
    }
    cout << (ok ? "YES" : "NO") << "\n";
    return 0;
}

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