Submission #945305

#	Time	Username	Problem	Language	Result	Execution time	Memory
945305		nguyennh	Bank (IZhO14_bank)	C++14	71 / 100	1068 ms	3640 KiB

bank

#include<bits/stdc++.h>
#define el '\n'
using namespace std ;

mt19937 rd(chrono::steady_clock::now().time_since_epoch().count());

int a[25] , b[25] , n , m;

namespace buffalo{
  bool can = false;
  
  void backtrack(int pos , int mask , bool check){
    if (!check) return;
    if (pos > n - 1){
      if (check) can = true;
      return;
    }
    for ( int sub_mask = 0 ; sub_mask < (1 << m) ; sub_mask++ ){
      int sum = 0;
      for ( int j = 0 ; j < m ; j++ ){
        if ((mask >> j & 1) || (sub_mask >> j & 1)) continue;
        sum += b[j];
      }
      if (sum == a[pos]){
        backtrack(pos + 1 , mask | ~sub_mask , true);
      }
      else backtrack(pos + 1 , sub_mask | mask , false);
    }
  }
  
  void solve(){
    backtrack(0 , 0 , 1);
    cout << (can ? "YES" : "NO");
  }
}

namespace sub_final{
  void solve(){
    vector<vector<int>> pre_mask(n + 5);
    for ( int i = 0 ; i < n ; i++ ){
      for ( int mask = 0 ; mask < (1 << m) ; mask++ ){
        int sum = 0;
        for ( int j = 0 ; j < m ; j++ ) sum += ((mask >> j & 1) ? b[j] : 0);
        if (sum == a[i]) pre_mask[i].push_back(mask);
      }
    }
    vector<int> store;
    bool satis = true;
    for ( int i = 0 ; i < n ; i++ ){
      if (i == 0){
        for ( auto x : pre_mask[i] ) store.push_back(x);
        if (store.empty()){
          satis = false;
          break;
        }
        continue;
      }
      if (store.empty()){
        satis = false;
        break;
      }
      vector<int> cur;
      for ( int j = 0 ; j < store.size() ; j++ ){
        for ( auto mask : pre_mask[i] ){
          if (mask & store[j]) continue;
          else cur.push_back(store[j] | mask);
        }
      }
      if (cur.empty()){
        satis = false;
        break;
      }
      swap(store , cur);
    }
    cout << (satis ? "YES" : "NO");
  }
}

int32_t main (){
  ios_base::sync_with_stdio(0);
  cin.tie(0);
  cin >> n >> m;
  for ( int i = 0 ; i < n ; i++ ) cin >> a[i];
  for ( int i = 0 ; i < m ; i++ ) cin >> b[i];
  if (n == 1){
    buffalo::solve();
    return 0;
  }
  sub_final::solve();
}

Compilation message (stderr)

bank.cpp: In function 'void sub_final::solve()':
bank.cpp:63:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   63 |       for ( int j = 0 ; j < store.size() ; j++ ){
      |                         ~~^~~~~~~~~~~~~~

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