Submission #1198605

#TimeUsernameProblemLanguageResultExecution timeMemory
1198605aram_7은행 (IZhO14_bank)C++20
100 / 100
512 ms80492 KiB
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);

    
    int n,m;
    cin>>n>>m;
    vector<int>a(n);
    vector<int>b(m);
    vector<vector<bool>>dp(1<<m,vector<bool>(n));
    for (int i = 0; i < n; i++){
        cin >> a[i];
    }
    for (int i = 0; i < m; i++){
        cin >> b[i];
    }
    vector<vector<int>> cnt(1001);
    for (int i = 1; i < (1 << m); i++) {
        int sum = 0;
        for (int j = 0; j < m; j++) {
            if (i & (1 << j)) {
                sum += b[j];
            }
        }
        if (sum <= 1000){
            cnt[sum].push_back(i);
        }
    }
    for (auto i : cnt[a[0]]) {
        dp[i][0] = 1;
    }
    for (int i = 0; i < (1 << m); i++) {
        for (int j = 0; j < n - 1; j++) {
            if (dp[i][j] == 0){
                continue;
            }
            for (int k : cnt[a[j + 1]]) {
                if (i & k){
                    continue;
                }
                dp[i + k][j + 1] = 1;
            }
        }
        if (dp[i][n - 1] == 1) {
            cout << "YES" << endl;
            return 0;
        }
    }
    cout << "NO" << endl;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...