This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include<bits/stdc++.h>
using namespace std;
typedef vector<int> vi;
typedef vector<vi> vvi;
//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];
vector<vi> coins_to_persons(1001, vector<int>());
vector<vi> persons_to_combs(n, vector<int>());
for (int i = 0; i < n; ++i) {
cin >> people[i];
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;
vvi dp(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]){
persons_to_combs[pers].push_back(i);
//cout << ans <<" " <<i << " " << pers << endl;
}
}
for(int i: persons_to_combs[0]){
dp[0][i] = 1;
if(n == 1) {
cout << "YES" << endl;
return 0;
}
}
bool ans = false;
//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]){
int pos = 1;
for (int i = 0; i < m; ++i){
if(comb & (1 << i)){
if(!(mask & (1 << i))) {
pos = 0;
break;
}
}
}
if(pos) dp[who][mask] = max(dp[who][mask], dp[who-1][mask^comb]);
//cout << who-1 << " " << comb << " "<< mask << " " << dp[who][mask]<< endl;
}
//cout << who << " " << mask << " " << dp[who+1][mask] << endl;
if((who == n-1) and dp[who][mask]) {
ans = true;
cout << "YES" << endl;
return 0;
//cout << who << " " << mask << endl;
}
}
}
if(ans) cout << "YES" << endl;
else cout << "NO" << endl;
return 0;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |