Submission #575411

#	Time	Username	Problem	Language	Result	Execution time	Memory
575411		ddy888	Bank (IZhO14_bank)	C++17	100 / 100	284 ms	16824 KiB

bank

#undef _GLIBCXX_DEBUG
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define pb push_back
#define fi first
#define si second
typedef pair<int,int> pi;
typedef tuple<int,int,int> ti;  
void debug_out() {cerr<<endl;}
template <typename Head, typename... Tail>
void debug_out(Head _H, Tail... _T) {cerr<<" "<<to_string(_H);debug_out(_T...);}
#define debug(...) cerr<<"["<<#__VA_ARGS__<<"]:",debug_out(__VA_ARGS__)
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

int n, m;
int ar[21], br[21];
int dp[(1 << 21) + 1];
int sum[(1 << 21) + 1];
int ans;

int main() {
    ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0); 

    cin >> n >> m;
    for (int i = 1; i <= n; ++i) cin >> ar[i], ar[i] += ar[i - 1];
    for (int i = 1; i <= m; ++i) cin >> br[i];
    for (int i = 1; i <= (1 << (m + 1)) - 1; ++i) {
        int s = 0;
        for (int j = 1; j <= m; ++j) {
            if (i & (1 << j)) s += br[j];
        }
        sum[i] = s;
    }

    memset(dp, -1, sizeof dp);
    dp[0] = 0;
    for (int now = 1; now <= (1 << (m + 1)) - 1; ++now) {
        for (int i = 1; i <= m; ++i) {
            if (!(now & (1 << i))) continue;

            int from = now ^ (1 << i);
            if (dp[from] == -1) continue;
            if (sum[now] == ar[dp[from] + 1]) {
                dp[now] = max(dp[now], dp[from] + 1);
            } else if (sum[now] < ar[dp[from] + 1]) {                
                dp[now] = max(dp[now], dp[from]);
            }
            ans = max(ans, dp[now]);
        }
    }

    cout << (ans >= n ? "YES" : "NO");
    return 0;
}

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