Submission #656013

#	Time	Username	Problem	Language	Result	Execution time	Memory
656013		iskhakkutbilim	Bank (IZhO14_bank)	C++14	44 / 100	1077 ms	416 KiB

bank

#include <bits/stdc++.h>
using namespace std;
 
#define int long long
#define ff first
#define ss second
#define all(a) a.begin(), a.end()
const int M = 1e9 + 7;
const int N = 1e6 + 1;
 
main(){
   ios::sync_with_stdio(0);
   cin.tie(0); cout.tie(0);
	int n, m; cin >> n >> m;
	int a[n], b[m];
	vector<int> g[n];
	for(int i = 0;i < n; i++){
		cin >> a[i];
	}
	for(int i = 0;i < m; i++){
		cin >> b[i];
	}
	sort(b, b + m);
	for(int mask = 0; mask < (1<<m); mask++){
		int sum = 0;
		for(int i = 0;i < m; i++){
			if(mask & (1<<i)){
				sum+= b[i];
			}
		}
		for(int i = 0;i < n; i++){
			if(a[i] == sum){
				g[i].push_back(mask);
			}
		}
	}
	for(int i = 0;i < n; i++){
		if(g[i].empty()){
			cout << "NO";
			return 0;
		}
	}
	if(n == 1){
		cout << "YES";
	}else{
		do{
			int cur_sum = 0, idx = 0;
			for(int i = 0;i < m; i++){
				if(idx >= n) break;
				cur_sum+= b[i];
				if(cur_sum == a[idx]){
					idx++, cur_sum = 0;
				}
			}
			if(idx >= n){
				cout << "YES";
				return 0;
			}
		}while(next_permutation(b, b + m));
		cout << "NO";
	}
	
	return 0;
}

Compilation message (stderr)

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

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