Submission #1229861

#	Time	Username	Problem	Language	Result	Execution time	Memory
1229861		escobrand	Bank (IZhO14_bank)	C++20	71 / 100	1097 ms	12252 KiB

bank

#include <bits/stdc++.h>
 
using namespace std;
#define all(v) v.begin(),v.end()
#define eb emplace_back
#define ll long long 
#define fi first
#define se second
int i,n,t,m,j,tmp;
int a[22],b[22],sum[(1<<20)],mask,sub;
bool dp[22][(1<<20)],ans;
vector<int> id[1000 * 20 + 10];
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    
    cin>>n>>m;
    for(i = 1;i<=n;i++)
    {
      cin>>a[i];
    }
    for(i = 0;i<m;i++)
    {
      cin>>b[i];
    }
    for( mask = 1;mask < (1<<m);mask++)
    {
      i = __builtin_ctz(mask);
      sum[mask] = sum[mask ^ (1<<i)] + b[i];
      id[sum[mask]].eb(mask);
    }
    
    dp[0][0] = 1;
    for(i = 1;i<=n-1;i++)
    {
      for( mask = 0;mask<(1<<m);mask++)
      {
        for(int k :id[a[i]])
        {
          if((k & mask) == k && dp[i-1][mask ^ k])
          {
            dp[i][mask] = 1;
            break;
          }
        }
      }
    }
    
    for( mask = 0;mask<(1<<m);mask++)
      {
        for(int k :id[a[i]])
        {
          if((k & mask) == k && dp[n-1][mask ^ k])
          {
            dp[n][mask] = 1;
            break;
          }
        }
        if(dp[n][mask])
        {
          cout<<"YES";
          return 0;
        }
      }
    cout<<"NO";
    
    return 0;
}

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