Submission #982856

#	Time	Username	Problem	Language	Result	Execution time	Memory
982856		dpsaveslives	Bank (IZhO14_bank)	C++17	100 / 100	738 ms	11088 KiB

bank

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 21;
int main()
{
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    int N,M; cin >> N >> M;
    vector<int> arr(N),bank(M);
    for(int i = 0;i<N;++i) cin >> arr[i];
    for(int i = 0;i<M;++i) cin >> bank[i];
    vector<vector<int>> poss(1001);
    for(int i = 1;i<(1<<M);++i){
        int cur = 0;
        for(int j = 0;j<M;++j){
            if((1<<j)&i){
                cur += bank[j];
            }
        }
        if(cur <= 1000) poss[cur].push_back(i);
    }
    vector<int> dp((1<<M),-1); dp[0] = 0;
    for(int i = 0;i<N;++i){
        for(int j = (1<<M)-1;j>=0;--j){
            if(dp[j] == -1) continue;
            for(auto t : poss[arr[i]]){
                if(!(t&j)){
                    dp[t|j] = dp[j]+1;
                    if(dp[t|j] == N){
                        cout << "YES\n";
                        return 0;
                    }
                }
            }
        }
    }
    cout << "NO\n";
    return 0;
}

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