Submission #805880

#TimeUsernameProblemLanguageResultExecution timeMemory
805880PixelCatPacking Biscuits (IOI20_biscuits)C++14
44 / 100
1094 ms972 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];
int val[MAXK + 10];
int sum[MAXK + 10][MAXK + 10];
vector<pii> trans[MAXK + 10];  // {minimum #tokens, bit}
int ans;

inline void dfs(int hibit, int cur, int k, int x) {
    for(auto &i:trans[hibit]) {
        if(i.F > cur) break;
        dfs(i.S + 1, cur - i.F, k, x);
    }
    cur = ((cur + sum[hibit][k - 1]) >> k);
    ans += cur / x + 1;
    // ans++;
    // while(cur >= x) {
    //     ans++; cur -= x;
    // }
}
 
long long count_tastiness(long long x, std::vector<long long> _a) {
    int k = sz(_a);

    memset(a, 0, sizeof(a));
    memset(val, 0, sizeof(val));
    memset(sum, 0, sizeof(sum));
    For(i, 0, k) trans[i].clear();
    
    For(i, 0, k - 1) {
        a[i] = _a[i];
        val[i] = a[i] << i;
    }
    For(i, 0, k) {
        sum[i][i] = val[i];
        For(j, i + 1, k) sum[i][j] = sum[i][j - 1] + val[j];
    }
    For(b1, 0, k - 1) {
        For(b2, b1, k - 1) {
            if((1ll << 60) / x < (1ll << b2)) break;
            trans[b1].eb((x << b2) - sum[b1][b2], b2);
        }
        sort(all(trans[b1]));
    }
    
    ans = 0;
    dfs(0, 0, k, x);
    return ans;
}
 
#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...