Submission #935972

#	Time	Username	Problem	Language	Result	Execution time	Memory
935972		Bula	Bank (IZhO14_bank)	C++17	100 / 100	109 ms	16976 KiB

bank

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int mod = 1e9 + 7, N = 2e5 + 5;

main(){
    ios::sync_with_stdio(0);
    cin.tie(0),cout.tie(0);
    int tt = 1; //cin >> tt;
    while(tt--){
        int n, m;
		cin >> n >> m;
		vector<int> a(n), b(m);
		for(int i = 0; i < n; i++) cin >> a[i];
		for(int i = 0; i < m; i++) cin >> b[i];

		vector<int> ans((1 << m), -1), left((1 << m), -1);
		ans[0] = left[0] = 0;
		for(int i = 0; i < (1 << m); i++){
			for(int j = 0; j < m; j++){
				if(!(i & (1 << j))) continue;
				int p = i & ~(1 << j);
				if(ans[p] == -1) continue;
				int sum = left[p] + b[j];
				int need = a[ans[p]];
				if(sum < need){
					ans[i] = ans[p];
					left[i] = sum;
				}else if(sum == need){
					ans[i] = ans[p] + 1;
					left[i] = 0;
				}
			}
			if(ans[i] == n){
				cout << "YES" << endl;
				return 0;
			}
		}
		cout << "NO" << endl;
    }
}

Compilation message (stderr)

bank.cpp:6:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
    6 | main(){
      | ^~~~

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