Submission #521729

#	Time	Username	Problem	Language	Result	Execution time	Memory
521729		Kalashnikov	Bank (IZhO14_bank)	C++17	71 / 100	1032 ms	10168 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 = 2e5+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;
		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]) {
				for(auto to: v) {
					if((to.F&mask) == 0) {
						cur.push_back({to.F|mask , to.S+1});
						if(to.S+1 == n) {
							cout << "YES";
							return;
						}
					}
				}
			}
		}
		for(auto to: cur) v.push_back(to);
	}
	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:35:22: warning: suggest parentheses around '-' inside '<<' [-Wparentheses]
   35 |     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:57:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   57 | main() {
      | ^~~~

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