Submission #636021

#	Time	Username	Problem	Language	Result	Execution time	Memory
636021		a_aguilo	Bank (IZhO14_bank)	C++17	19 / 100	119 ms	8404 KiB

bank

#include<bits/stdc++.h>

using namespace std;
typedef vector<int> vi;
typedef vector<vi> vvi;

vvi dp;
vvi possibilities;

//dp[i][j] = es posible satisfacer a los clientes i con j monedas



int main(){
    int n, m;
    cin >> n >> m;
    int people[n];
    int coins[m];
    map<int, vi> coins_to_persons;
    map<int, vi> persons_to_combs;
    for (int i = 0; i < n; ++i) {
        cin >> people[i];
        if(coins_to_persons.find(people[i]) == coins_to_persons.end()) coins_to_persons[people[i]] = {i};
        else coins_to_persons[people[i]].push_back(i);
    }
    //cout << " <" << endl;
    for(int i = 0; i < m; ++i){
        cin >> coins[i];
        //cout << i << " " << coins[i] << endl;
    }
    //cout << " << " << endl;
    dp = vvi(n, vi(1<<m, 0));
    //cout << " a " << endl;
    for(int i = 0; i < (1 << m); ++i){
        //cout << "aa" << endl;
        //cout << i << endl;
        int ans = 0;
        for(int j = 0; j < m; ++j){
            if(i&(1 << j)) ans+= coins[j];
        }
        //cout << ans << endl;
        for(int pers: coins_to_persons[ans]){
            if(persons_to_combs.find(pers) == persons_to_combs.end()) persons_to_combs[pers] = {i};
            else persons_to_combs[pers].push_back(i);
            //cout << ans <<" " <<i << " " << pers << endl;
        }
    }
    bool ans = false;
    for(int combo: persons_to_combs[0]){
        dp[0][combo] = 1;
        if(n == 1) ans = true;
    }
    //cout << "a" << endl;
    for(int who = 1; who < n; ++who){
        for(int mask = 0; mask < (1 << m); ++mask){
            for(int comb : persons_to_combs[who]){
                if(comb& mask == comb) dp[who][mask] = max(dp[who][mask], dp[who-1][mask^comb]);
            }
            if(who == n-1 and dp[who][mask]) ans = true;
        }

    }
    if(ans) cout << "YES" << endl;
    else cout << "NO" << endl;
    return 0;
}

Compilation message (stderr)

bank.cpp: In function 'int main()':
bank.cpp:57:31: warning: suggest parentheses around comparison in operand of '&' [-Wparentheses]
   57 |                 if(comb& mask == comb) dp[who][mask] = max(dp[who][mask], dp[who-1][mask^comb]);
      |                          ~~~~~^~~~~~~

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