제출 #683106

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
683106		brijeshSiwach	은행 (IZhO14_bank)	C++17	0 / 100	78 ms	172748 KiB

bank

//----------------------- JAI BABA KI -----------------------//    

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef long double ld;
typedef vector<int> vi;
typedef vector<ll> vl;
typedef pair<ll,ll> pl;
typedef vector<pl> vpl;
 
#define all(x) x.begin(), x.end()
#define nl cout<<"\n"
#define ar  array
#define PRECISION 9
#define fast_IO ios_base::sync_with_stdio(false),cin.tie(NULL),cout.tie(NULL)
#define clock() cerr << "Time elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC << " s.\n"

#ifdef LOCAL
#include "../Library/debug.h"
#else 
#define dbg(...) "OM BHOLE"
#endif


//----------------------- ये कर लो पहले -----------------------//
const int mxN=1<<21;
int n, m, a[mxN], b[mxN], dp[21][mxN];
unordered_map<int,vector<int>> maskp; 

int go(int pos, int mask){
    if(pos==n) return 1;
    if(dp[pos][mask]==-1){
        int ans=0;
        for(int smask: maskp[a[pos]]){
            if(mask&smask) continue;
            ans|=go(pos+1,mask|smask);
        }
        dp[pos][mask]=ans;
    }
    return dp[pos][mask];
}

void solve(){
    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 i=0;i<1<<m;i++){
        int sum=0;
        for(int j=0;j<m;j++){
            if(i&(1<<j)) sum+=b[j];
        }
        maskp[sum].push_back(i);
    }
    memset(dp,-1,sizeof(dp));
    cout<<(go(0,0)==1?"YES\n":"NO\n");
}

//----------------------- पढाई कर ले -----------------------//

int main(){
    if (0) cout<<fixed<<setprecision(PRECISION);
    fast_IO;
    int T=1;
    // cin>>T;
    freopen("bank.in", "r",stdin);
    freopen("bank.out", "w",stdout);
    while(T--){
        solve();
        clock();
    }
    return 0;
}

컴파일 시 표준 에러 (stderr) 메시지

bank.cpp: In function 'int main()':
bank.cpp:71:12: 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:12: 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: 0 / 19
• Subtask 2: 0 / 25
• Subtask 3: 0 / 27
• Subtask 4: 0 / 29