Submission #362735

#	Time	Username	Problem	Language	Result	Execution time	Memory
362735		sumit_kk10	Bank (IZhO14_bank)	C++14	0 / 100	50 ms	48236 KiB

bank

#include<bits/stdc++.h>
#define fast ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
using namespace std;
const int MOD = 1e9 + 7;
const int N = 1e6 + 2;
vector<int> pos[N];
int n, m, a[N], b[N];
bool vis[N];

long long gen(int i, long long sum, vector<int>& cur){
    if(i == m){
        long long power = 0;
        for(auto k : cur)
            power = (power | k);
        pos[sum].push_back(power);
        return 0;
    }
    vector<int> cur1 = cur;
    gen(i + 1, sum, cur1);
    cur1.push_back(i);
    gen(i + 1, sum + b[i], cur1);
}

void solve() {
    cin >> n >> m;
    for(int i = 0; i < n; ++i)
        cin >> a[i];
    for(int i = 0; i < m; ++i)
        cin >> b[i];
    vector<int> ok;
    gen(0, 0, ok);
    for(int i = 0; i < n; ++i){
        long long sum = a[i];
        bool yes = 0;
        for(auto k : pos[sum]){
            bool no = true;
            for(int j = 0; j < m; ++j){
                if(k & (1LL << j)){
                    if(vis[j]){
                        no = 0;
                        break;
                    }
                }
            }
            if(no){
                yes = true;
                for(int j = 0; j < m; ++j)
                	if(k & (1LL << j))
                		vis[j] = true;
                break;
            }
        }
        if(!yes){
            cout << "NO\n";
            return;
        }
    }
    cout << "YES\n";
}

int main() {
    fast;
    int t = 1;
    // cin >> t;
    while(t--)
        solve();
    return 0;
}

Compilation message (stderr)

bank.cpp: In function 'long long int gen(int, long long int, std::vector<int>&)':
bank.cpp:18:24: warning: control reaches end of non-void function [-Wreturn-type]
   18 |     vector<int> cur1 = cur;
      |                        ^~~

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