Submission #805420

#TimeUsernameProblemLanguageResultExecution timeMemory
805420PixelCat비스킷 담기 (IOI20_biscuits)C++14
21 / 100
1083 ms63332 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];
// unordered_map<int, int> mem[MAXK + 10];
map<pii, int> mem;
 
// int dp(int k, int cur, int x) {
//     cur /= 2;
//     if(k >= MAXK) return cur + 1;
//     cur += a[k];
//     if(mem[k].count(cur)) return mem[k][cur];
//     return mem[k][cur] = dp(k + 1, cur, x) + (cur < x ? 0 : dp(k + 1, cur - x, x));
// }

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

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