# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
794963 | nguyennehehe | 은행 (IZhO14_bank) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include<bits/stdc++.h>
using namespace std;
const int N = 20;
int n, m, a[N], b[N];
void sub1() {
bool ok = false;
for (int s = 1; s < (1 << m); ++s) {
int sum = 0;
for (int i = 0; i < m; ++i) {
if (s >> i & 1) sum += b[i];
}
ok |= sum == a[0];
}
cout << (ok ? "YES" : "NO");
}
void sub2() {
bool good = false;
sort(b, b + m);
do {
bool ok = true;
for (int i = 0, j = 0; i < n; ++i) {
int sum = 0;
while (j < m && sum < a[i]) {
sum += b[j];
j += 1;
}
ok &= sum == a[i];
}
good |= ok;
if (good) break;
} while (next_permutation(b, b + m));
cout << (good ? "YES": "NO");
}
void maximize(int &x, int y) {
if (x < y) x = y;
}
void sub3() {
vector<int> sum(1 << m);
for (int s = 1; s < (1 << m); ++s) {
int cur = 0;
for (int i = 0; i < m; ++i) {
if (s >> i & 1) cur += b[i];
}
sum[s] = cur;
}
vector dp(n, vector<int>(1 << m));
for (int s = 0; s < (1 << m); ++s) {
if (sum[s] == a[0]) dp[0][s] = 1;
}
for (int i = 1; i < n; ++i) {
for (int s = 0; s < (1 << m); ++s) {
int msk = ((1 << m) - 1) ^ s;
for (int x = msk; ; x = (x - 1) & msk) {
if (sum[x] == a[i]) {
maximize(dp[i][s], dp[i - 1][x] + 1);
}
else {
dp[i][x] = dp[i - 1][x];
}
if (x == 0) break;
}
}
}
bool good = false;
for (int s = 0; s < (1 << m); ++s) {
if (dp[n - 1][s] == n) good = true;
}
cout << (good ? "YES": "NO");
}
signed main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
cin >> n >> m;
for (int i = 0; i < n; ++i) cin >> a[i];
for (int i = 0; i < m; ++i) cin >> b[i];
sub3();
}