Submission #362634

#	Time	Username	Problem	Language	Result	Execution time	Memory
362634		valerikk	Packing Biscuits (IOI20_biscuits)	C++17	42 / 100	1056 ms	552 KiB

biscuits

#include <bits/stdc++.h>
 
using namespace std; 
 
typedef long long ll;
 
ll count_tastiness(ll x, vector<ll> a) {
    const int k = 60;
    while ((int)a.size() < k + 1) a.push_back(0);
    if (x <= 10000) {
        for (int i = 0; i < k; i++) {
            ll z = max(0LL, a[i] - x) / 2;
            a[i + 1] += z;
            a[i] -= 2 * z;
        }
        const int C = x + 2;
        vector<ll> dp(C, 0);
        dp[0] = 1;
        for (int i = 0; i < k; i++) {
            vector<ll> ndp(C, 0);
            for (int carry = 0; carry < C; carry++) {
                assert(a[i] >= 0 && a[i] <= x + 1);
                int cur = carry + a[i];
                ndp[cur / 2] += dp[carry];
                if (cur >= x) ndp[(cur - x) / 2] += dp[carry];
            }
            dp.swap(ndp);
        }
        ll res = 0;
        for (const ll &z : dp) res += z;
        return res;
    }
    ll res = 0;
    for (ll y = 0; y <= 100000; y++) {
        auto a_copy = a;
        bool can = true;
        for (int i = 0; i < k; i++) {
            ll need = x * ((y >> i) & 1);
            if (a[i] < need) {
                can = false;
                break;
            }
            a[i] -= need;
            a[i + 1] += a[i] / 2;
        }
        if (can) res++;
        a = a_copy;
    }
    return res;
}
 
#ifdef LOCAL
int main() {
    freopen("input.txt", "r", stdin);
    ios::sync_with_stdio(false);
    cin.tie(0);
    int k;
    ll x;
    cin >> k >> x;
    vector<ll> a(k);
    for (ll &z : a) cin >> z;
    cout << count_tastiness(x, a);
    return 0;
}
#endif

• Subtask 1: 9 / 9
• Subtask 2: 12 / 12
• Subtask 3: 21 / 21
• Subtask 4: 0 / 35
• Subtask 5: 0 / 23