Submission #285303

#	Time	Username	Problem	Language	Result	Execution time	Memory
285303		BeanZ	Bank (IZhO14_bank)	C++14	52 / 100	1089 ms	20344 KiB

bank

#include <bits/stdc++.h>

using namespace std;

#define ll long long
#define endl '\n'

const int N = 2e5 + 5;
const int mod = 1e9 + 7;
vector<ll> choice[1005];
ll a[25], b[25], dp[2000006];
int main(){
        ios_base::sync_with_stdio(false);
        cin.tie(0);
        if (fopen("VietCT.INP", "r")){
                freopen("VietCT.INP", "r", stdin);
                freopen("VietCT.OUT", "w", stdout);
        }
        ll n, m;
        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 = 1; i < (1 << m); i++){
                ll cost = 0;
                for (int j = 0; j < m; j++){
                        if (i & (1 << j)){
                                cost += b[j];
                        }
                }
                if (cost <= 1000) choice[cost].push_back(i);
        }
        for (int i = 1; i < (1 << m); i++) dp[i] = -2;
        dp[0] = -1;
        bool flag = false;
        for (int i = 0; i < n; i++){
                for (int j = 1; j < (1 << m); j++){
                        if (choice[a[i]].size() == 0) continue;
                        for (auto k : choice[a[i]]){
                                if ((j & k) == k){
                                        if (dp[j ^ k] == (i - 1)){
                                                dp[j] = i;
                                        }
                                }
                        }
                        //cout << dp[j] << " " << j << endl;
                        if (dp[j] == (n - 1)) flag = true;
                }
        }
        if (flag) cout << "YES";
        else cout << "NO";
}
/*
*/

Compilation message (stderr)

bank.cpp: In function 'int main()':
bank.cpp:16:24: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)', declared with attribute warn_unused_result [-Wunused-result]
   16 |                 freopen("VietCT.INP", "r", stdin);
      |                 ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~
bank.cpp:17:24: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)', declared with attribute warn_unused_result [-Wunused-result]
   17 |                 freopen("VietCT.OUT", "w", stdout);
      |                 ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~

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