Submission #521844

#	Time	Username	Problem	Language	Result	Execution time	Memory
521844		Kalashnikov	Bank (IZhO14_bank)	C++17	71 / 100	1052 ms	4652 KiB

bank

#include <bits/stdc++.h>
 
#define ios ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0)
#define file(s) if (fopen(s".in", "r")) freopen(s".in", "r", stdin), freopen(s".out", "w", stdout)
#define all(a) a.begin() , a.end()
#define F first
#define S second
 
using namespace std;
using ll = long long;
 
const int N = 20+5 , inf = 2e9 + 7;
const ll INF = 1e18 ,   mod = 1e9+7 , P = 6547;

int a[N] , b[N];


void solve(int tc) {
	int n , m;
	cin >> n >> m;
	for(int i = 1; i <= n; i ++) {
		cin >> a[i];
	}
	for(int i = 1; i <= m; i ++) {
		cin >> b[i];
	}
	vector<pair<int,int>> v;
	v.push_back({0 , 0});
	for(int i = 1; i <= n; i ++) {
		int ok = 0;
		vector<pair<int,int>> cur;
		vector<int> ms;
		for(int mask = 0; mask < (1 << m); mask ++) {
			int sum = 0;
			for(int j = 1; j <= m; j ++) {
				if(mask & (1 << j-1)) {
					sum += b[j];
				}
			}
			if(sum == a[i]) {
				ms.push_back(mask);
			}
		}
		for(auto x: ms) {
			for(auto y: v) {
				if((x&y.F) == 0) {
					cur.push_back({x|y.F , y.S+1});
					if(y.S+1 == n) {
						cout << "YES\n";
						return;
					}
				}
			}
		}
		swap(cur , v);
	}
	cout << "NO";
}
/*
*/
main() {
    ios;
    int tt = 1 , tc = 0;
    // cin >> tt;
    while(tt --) {
        solve(++tc);
    }
    return 0;
}

Compilation message (stderr)

bank.cpp: In function 'void solve(int)':
bank.cpp:36:22: warning: suggest parentheses around '-' inside '<<' [-Wparentheses]
   36 |     if(mask & (1 << j-1)) {
      |                     ~^~
bank.cpp:30:7: warning: unused variable 'ok' [-Wunused-variable]
   30 |   int ok = 0;
      |       ^~
bank.cpp: At global scope:
bank.cpp:61:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   61 | main() {
      | ^~~~

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