Submission #805199

#TimeUsernameProblemLanguageResultExecution timeMemory
805199PixelCatPacking Biscuits (IOI20_biscuits)C++14
0 / 100
21 ms4052 KiB
#include "biscuits.h"

#ifdef NYAOWO
#include "grader.cpp"
#endif

#include <bits/stdc++.h>
#define For(i, a, b) for(int i = a; i <= b; i++)
#define Forr(i, a, b) for(int i = a; i >= b; i--)
#define F first
#define S second
#define all(x) x.begin(), x.end()
#define sz(x) ((int)x.size())
#define eb emplace_back
#define int LL
using namespace std;
using i32 = int32_t;
using LL = long long;
using pii = pair<int, int>;

const int MAXK = 60;

int a[MAXK + 10];

map<pii, int> mem;

int dp(int k, int x) {
    x /= 2;
    if(k < MAXK) x += a[k];
    pii key(k, x);
    if(mem.count(key)) return mem[key];
    if(k > MAXK && x == 0) return mem[key] = 1;
    return mem[key] = dp(k + 1, x) + (x == 0 ? 0 : dp(k + 1, x - 1));
}

long long count_tastiness_2(long long x, std::vector<long long> _a) {
    memset(a, 0, sizeof(a));
    int k = sz(_a);
    int su = 0;
    For(i, 0, k - 1) {
        a[i] = _a[i];
        su += (1ll << i) * a[i];
    }
    int ans = 0;
    For(s, 0, su) {
        int cur = 0;
        bool fail = false;
        For(i, 0, 19) {
            cur = cur / 2 + a[i];
            if(s & (1 << i)) cur -= x;
            if(cur < 0) {
                fail = true;
                break;
            }
        }
        if(!fail) ans++;
    }
    return ans;
}

long long count_tastiness(long long x, std::vector<long long> _a) {
    memset(a, 0, sizeof(a));
    int k = sz(_a);
    For(i, 0, k - 1) {
        a[i] = _a[i];
    }
    // cerr << count_tastiness_2(1, _a) << "\n";
    return dp(0, 0);
}

#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...