제출 #1001854

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1001854		coin_	은행 (IZhO14_bank)	C++14	100 / 100	99 ms	34480 KiB

bank

#include <bits/stdc++.h>
#define endl '\n'
#define int long long
#define fi first
#define se second
#define pb push_back
using namespace std;
signed main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    int n, m;
    cin >> n >> m;
    int a[n], b[m];
    for (int i = 0; i < n; i++){
        cin >> a[i];
    }
    for (int i = 0; i < m; i++){
        cin >> b[i];
    }
    pair<int, int> dp[(1 << m)];
    int visited[(1 << m)] = {0};
    dp[0] = {0, 0};
    visited[0] = 1;
    vector<vector<int>> masks(22);
    for (int i = 0; i < (1 << m); i++){
        masks[__builtin_popcount(i)].push_back(i);
    }
    for (int i = 0; i <= m; i++){
        for (int mask: masks[i]){
            if (visited[mask] == 0) continue;
            if (dp[mask].fi == n){
                cout << "YES";
                return 0;
            }
            int curperson = dp[mask].fi;
            for (int j = 0; j < m; j++){
                int newmask = mask | (1 << j);
                if (mask == newmask) continue;
                if (dp[mask].se + b[j] > a[curperson]) continue;
                visited[newmask] = 1;
                dp[newmask] = {curperson, dp[mask].se + b[j]};
                if (a[curperson] == dp[mask].se + b[j]){
                    dp[newmask] = {curperson+1, 0};
                }
            } 
        }
    }
    cout << "NO";
}

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