Submission #862791

#	Time	Username	Problem	Language	Result	Execution time	Memory
862791		Requiem	Bank (IZhO14_bank)	C++17	100 / 100	91 ms	16984 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];
int n,m;
ii dp[(1<<20)+9];
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++){
        dp[i] = ii(-1,-1);
    }
    dp[0] = {0,0};
    int ans = 0;
    for(int msk=0;msk<(1<<m);msk++){
        for(int j=m-1;j>=0;j--){
            if (msk & (1<<j)){
                int prevmsk = msk ^ (1<<j);
                int curr_budget = dp[prevmsk].se + b[j+1];
                if (dp[prevmsk].fi == -1) continue;
                int nxtp = a[dp[prevmsk].fi+1];
                // cout<<msk<<' '<<prevmsk<<' '<<dp[prevmsk].fi<<' '<<dp[prevmsk].se<<endl;

                if (curr_budget < nxtp){
                    dp[msk] = ii(dp[prevmsk].fi,curr_budget);
                }
                else if (curr_budget == nxtp){
                    dp[msk] = ii(dp[prevmsk].fi+1,0);
                }
            }
        }
        if (dp[msk].fi == n){
            cout<<"YES"<<endl;
            return 0;
        }
    }

    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:40:9: warning: unused variable 'ans' [-Wunused-variable]
   40 |     int ans = 0;
      |         ^~~

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