제출 #362634

#제출 시각아이디문제언어결과실행 시간메모리
362634valerikk비스킷 담기 (IOI20_biscuits)C++17
42 / 100
1056 ms552 KiB
#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
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...