Submission #1182735

#	Time	Username	Problem	Language	Result	Execution time	Memory
1182735		FaresSTH	Bank (IZhO14_bank)	C++20	100 / 100	390 ms	20536 KiB

bank

#include "bits/stdc++.h"
using namespace std;
int n,m,a[25],b[25];
bool dp[25][1<<20];
vector<int>s[1001];
int main(){
    cin.tie(0)->sync_with_stdio(0);
    cin>>n>>m;
    for(int i=0;i<n;i++)cin>>a[i];
    for(int i=0;i<m;i++)cin>>b[i];
    for(int mask=0;mask<(1<<m);mask++){
        int sum=0;
        for (int j=0;j<m;j++){
            if(mask&(1<<j))sum+=b[j];
        }
        if(sum<=1000)s[sum].push_back(mask);
    }
    dp[0][0]=1;
    for(int i=1;i<=n;i++){
        for(int mask=0;mask<(1<<m);mask++){
            if(!dp[i-1][mask])continue;
            for(int k:s[a[i-1]]){
                if((mask&k))continue;
                dp[i][(mask|k)]=1;
            }
        }
    }
    cout<<(*max_element(dp[n],dp[n]+(1<<m))?"YES":"NO");
}

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