#pragma GCC optimize("Ofast")
#include <bits/stdc++.h>
using namespace std;
#define el "\n"
#define pii pair<int, int>
#define pll pair<ll, ll>
#define fi first
#define se second
#define mp make_pair
#define sqr(x) ((x) * (x))
#define FOR(i, l, r) for (int i = l; i <= (r); i++)
#define FOD(i, l, r) for (int i = l; i >= (r); i--)
#define all(x) begin(x), end(x)
#define rall(x) rbegin(x), rend(x)
#define sz(x) ((int)(x).size())
#define fast_io ios_base::sync_with_stdio(false); cin.tie(nullptr);
using db = long double;
using ll = long long;
using ull = unsigned long long;
using vi = vector<int>;
using vll = vector<ll>;
using vpii = vector<pii>;
using vpll = vector<pll>;
using vvi = vector<vi>;
using vvll = vector<vll>;
using vbool = vector<bool>;
using vvbool = vector<vbool>;
template<class T> inline bool chmax(T &a, T const &b) { return (a < b ? (a = b, true) : false); }
template<class T> inline bool chmin(T &a, T const &b) { return (a > b ? (a = b, true) : false); }
// #define DEBUG
#ifdef DEBUG
#include "D:\cpp\debug.h"
#else
#define debug(...)
#define debug_arr(...)
#endif
mt19937 rng((uint32_t)chrono::steady_clock::now().time_since_epoch().count());
constexpr int N = 20;
constexpr int INF = 1E9 + 7;
constexpr ll INFLL = 4E18;
constexpr int MOD = 1E9 + 7; // 998244353
constexpr double EPS = 1E-10;
#define bit(i, mask) ((mask >> i) & 1)
bool dp[N + 1][1 << N];
int sum[1 << N];
int a[N + 1], b[N];
int n, m;
void solve() {
cin >> n >> m;
int salary_sum = 0;
FOR(i, 1, n) {
cin >> a[i];
salary_sum += a[i];
}
int banknote_sum = 0;
FOR(i, 0, m - 1) {
cin >> b[i];
banknote_sum += b[i];
}
if (salary_sum > banknote_sum) {
cout << "NO" << el;
return;
}
const int full_mask = (1 << m) - 1;
FOR(mask, 0, full_mask) {
dp[0][mask] = true;
FOR(i, 0, m - 1) {
if (bit(i, mask)) {
sum[mask] += b[i];
}
}
}
FOR(i, 1, n) {
FOR(mask, 0, full_mask) {
if (dp[i][mask]) continue;
if (sum[mask] == a[i] && dp[i - 1][full_mask & ~mask]) {
for (int overmask = mask; overmask <= full_mask; overmask = (overmask + 1) | mask) {
dp[i][overmask] = true;
}
}
}
}
cout << (dp[n][full_mask] ? "YES" : "NO") << el;
}
int main() {
fast_io
#define LOCAL
#ifndef LOCAL
#define PROBLEM "bank"
freopen(PROBLEM ".in", "r", stdin);
freopen(PROBLEM ".out", "w", stdout);
#endif
int t = 1;
// cin >> t;
while (t--) solve();
}
| # | 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... |