Submission #862341

#	Time	Username	Problem	Language	Result	Execution time	Memory
862341		Requiem	Bank (IZhO14_bank)	C++17	52 / 100	1050 ms	30144 KiB

bank

#include<bits/stdc++.h>
#define int long long
#define pb push_back
#define fast ios_base::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
#define MOD 1000000007
#define INF 1e18
#define fi first
#define se second
#define FOR(i,a,b) for(int i=a;i<=b;i++)
#define FORD(i,a,b) for(int i=a;i>=b;i--)
#define sz(a) ((int)(a).size())
#define pi 3.14159265359
#define TASKNAME "bank"
template<typename T> bool maximize(T &res, const T &val) { if (res < val){ res = val; return true; }; return false; } 
template<typename T> bool minimize(T &res, const T &val) { if (res > val){ res = val; return true; }; return false; } 
using namespace std; 
typedef pair<int,int> ii; 
typedef pair<int,ii> iii; 
typedef vector<int> vi; 
const int MAXN = 23;
int a[MAXN],b[MAXN],value[(1<<20)+5];
int n,m;
bool dp[21][(1<<20)+5];
main() 
{ 
    fast; 
    // freopen(TASKNAME".inp","r",stdin); 
    // freopen(TASKNAME".out","w",stdout); 
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    for(int i=1;i<=m;i++){
        cin>>b[i];
    }
    for(int i=0;i<(1<<m);i++){
        int sum = 0;
        for(int j=0;j<m;j++){
            if (i & (1LL<<j)) value[i] += b[j+1];
        }
    }
    memset(dp,false,sizeof(dp));
    dp[0][0] = true;
    bool check = false;
    for(int i=1;i<=n;i++){
        for(int mask=0;mask<(1<<m);mask++){
            // cout<<"MASK: "<<mask<<endl;
            for(int submask=mask;submask;submask = (submask-1) & mask){
                // cout<<submask<<endl;
                if (value[submask] == a[i]) dp[i][mask] |= dp[i-1][mask ^ submask];
            }
            if (i==n) check |= dp[i][mask];
        }
    }
    if (check) cout<<"YES"<<endl;
    else cout<<"NO"<<endl;
    
    
} 

Compilation message (stderr)

bank.cpp:24:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   24 | main()
      | ^~~~
bank.cpp: In function 'int main()':
bank.cpp:37:13: warning: unused variable 'sum' [-Wunused-variable]
   37 |         int sum = 0;
      |             ^~~

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