제출 #345822

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
345822		acm	은행 (IZhO14_bank)	C++14	71 / 100	1073 ms	34620 KiB

bank

#include <bits/stdc++.h>
#define speed                   \
  ios_base::sync_with_stdio(0); \
  cin.tie(0);                   \
  cout.tie(0);
#define precision     \
  cout.precision(30); \
  cerr.precision(10);
#define ll long long
#define ld long double
#define pll pair<ll, ll>
#define pii pair<int, int>
#define forn(n) for (int i = 1; i <= n; i++)
#define forlr(l, r) for (int i = l; i != r; (l > r ? i-- : i++))
#define pb(x) push_back(x)
#define sz(x) (int)x.size()
#define mp(x, y) make_pair(x, y)
#define all(x) x.begin(), x.end()
#define pc(x) __builtin_popcount(x)
#define pcll(x) __builtin_popcountll(x)
#define F first
#define S second
using namespace std;
int n, m, a[25], b[25], c[(1 << 20)];
bitset<(1 << 20)> dp[25];
vector<int> g[1000005];
int main() {
  speed;
  precision;
  // code
  cin >> n >> m;
  for (int i = 1; i <= n; i++) cin >> a[i];
  for (int i = 0; i < m; i++) cin >> b[i];
  for (int i = 1; i < (1 << m); i++)
    c[i] = c[i ^ (1 << __lg(i))] + b[__lg(i)], g[c[i]].pb(i);
  if (n == 1) {
    cout << (sz(g[a[1]]) ? "YES\n" : "NO\n");
    return 0;
  }
  dp[0][0] = 1;
  for (int i = 1; i <= n; i++) {
    int ok = 0;
    for (int k = 0; k < (1 << m); k++) {
      if ((1 << pc(k)) < sz(g[a[i]])) {
        for (int l = k; l; l = (l - 1) & k) {
          if (c[l] == a[i] && dp[i - 1][k ^ l]) {
            dp[i][k] = 1;
            ok = 1;
            break;
          }
        }
      } else {
        for (auto l : g[a[i]]) {
          if (l > k) break;
          if ((k & l) == l && dp[i - 1][k ^ l]) {
            dp[i][k] = 1;
            ok = 1;
            break;
          }
        }
      }
    }
    if (!ok) {
      cout << "NO\n";
      return 0;
    }
  }
  cout << "YES\n";
  // endl
#ifndef ONLINE_JUDGE
  cerr << "\nTime elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC << " s.\n";
#endif
  return 0;
}

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