#define nametask ""
#include <bits/stdc++.h>
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define SZ(a) (int) a.size()
#define all(a) a.begin(), a.end()
#define FOR(i, a, b) for (int i = a; i <= b; i++)
#define FOD(i, b, a) for (int i = b; i >= a; i--)
using namespace std;
typedef long long ll;
typedef pair <int, int> pi;
const int N = 20;
int n, m;
int a[N], b[N];
int sum[(1 << N)];
bool f[(1 << N)][21];
signed main()
{
ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
if (fopen(nametask".inp", "r")){
freopen(nametask".inp", "r", stdin);
freopen(nametask".out", "w", stdout);
}
cin >> n >> m;
FOR(i, 0, n - 1) cin >> a[i];
FOR(i, 0, m - 1) cin >> b[i];
FOR(mask, 0, (1 << m) - 1){
FOR(i, 0, m - 1){
if ((mask >> i) & 1) sum[mask] += b[i];
}
}
FOR(mask, 0, (1 << m) - 1){
if (sum[mask] == a[0]) f[mask][0] = 1;
}
bool ck = 0;
FOR(i, 1, n - 1){
FOR(mask, 1, (1 << m) - 1){
for (int smask1 = mask; smask1 > 0; smask1 = (smask1 - 1) & mask){
int smask2 = mask ^ smask1;
if (sum[smask1] == a[i]){
f[mask][i] |= f[smask2][i - 1];
}
}
}
}
FOR(mask, 0, (1 << m) - 1) {
if (f[mask][n - 1]) {
cout << "YES\n";
return 0;
}
}
cout << "NO\n";
return 0;
}
컴파일 시 표준 에러 (stderr) 메시지
bank.cpp: In function 'int main()':
bank.cpp:26:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
26 | freopen(nametask".inp", "r", stdin);
| ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~
bank.cpp:27:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
27 | freopen(nametask".out", "w", stdout);
| ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~
# | 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... |