Submission #1302162

#	Time	Username	Problem	Language	Result	Execution time	Memory
1302162		nguthianmangcay	Bank (IZhO14_bank)	C++20	100 / 100	344 ms	55080 KiB

bank

#include<bits/stdc++.h>
using namespace std;
const int N=22;
const int MAS = (1<<22);
const int VL = 1e3+3;
const long long inf=1e18+3;
#define ll long long
#define fi first
#define se second
#define VOI void
#define MASK(i) (1LL<<i)
vector<int>premask[VL];

int dp[N][MAS];

int a[N],b[N];

VOI jiangly(){

    int n,m;
    cin>>n>>m;
    int maxa = 0;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        maxa = max(maxa,a[i]);
    }
    for(int i=1;i<=m;i++){
        cin>>b[i];
    }
    int tt = MASK(m) - 1;
    for(int mask=0;mask<=tt;mask++){
        int sum = 0;
        for(int i=1;i<=m;i++){
            if(mask & MASK(i-1)){
                sum += b[i];
            }
        }
        if(sum <= maxa){
            premask[sum].push_back(mask);
        }
    }
    dp[0][0] = 1;
    for(int i=0;i<n;i++){
        for(int mask=0;mask<=tt;mask++){
            if(dp[i][mask]){
                int nxt = i+1;
                int nxtval = a[i+1];
                for(int add : premask[nxtval]){
                    if(!(mask & add)){
                        int nmask = mask | add;
                        dp[nxt][nmask] |= dp[i][mask];
                    }
                }
            }
        }
    }
    int res = 0;
    for(int mask=0;mask<=tt;mask++){
        res = max(res,dp[n][mask]);
    }
    if(res)cout<<"YES";
    else cout<<"NO";

}
int main(){
    cin.tie(0)->sync_with_stdio(0);
    if(fopen("QUANSENSEI.inp","r")){
        freopen("O(0).inp","r",stdin);
    }
    if(fopen("bank.in","r")){
        freopen("bank.in","r",stdin);
        freopen("bank.out","w",stdout);
    }

//    if(fopen("input.txt","r")){
//        freopen("input.txt","r",stdin);
//        freopen("output.txt","w",stdout);
//    }
    jiangly();

//    cerr << "Time elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC << " s.\n";
}

Compilation message (stderr)

bank.cpp: In function 'int main()':
bank.cpp:68:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   68 |         freopen("O(0).inp","r",stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~
bank.cpp:71:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   71 |         freopen("bank.in","r",stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~
bank.cpp:72:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   72 |         freopen("bank.out","w",stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~

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