Submission #397168

#	Time	Username	Problem	Language	Result	Execution time	Memory
397168		rocks03	Packing Biscuits (IOI20_biscuits)	C++14	9 / 100	1092 ms	124624 KiB

biscuits

//#pragma GCC target("avx2")
//#pragma GCC optimization("O3")
//#pragma GCC optimization("unroll-loops")
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pii pair<int, int>
#define pll pair<ll, ll>
#define ff first
#define ss second
#define pb push_back
#define SZ(x) ((int)(x).size())
#define all(x) x.begin(), x.end()
#define rep(i, a, b) for(int i = (a); i < (b); i++)
#define per(i, a, b) for(int i = (a); i >= (b); i--)
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

int N;
ll x;
vector<ll> a;

map<ll, ll> memo[100];
ll f(ll y, int i){
    if(memo[i].count(y)){
        return memo[i][y];
    }
    ll& ans = memo[i][y];
    ans = 1;
    rep(j, i + 1, N){
        y += (1ll << j);
        if(a[j] / x >= y){
            ans += f(y, j);
        }
        y -= (1ll << j);
    }
    return ans;
}

long long count_tastiness(long long X, vector<long long> A){
    while(SZ(A) < 62) A.pb(0);
    a = A, x = X, N = SZ(A);
    rep(i, 0, N){
        a[i] = a[i] * (1ll << i);
        if(i) a[i] += a[i - 1];
    }
    rep(i, 0, N) memo[i].clear();
    ll ans = 1;
    rep(i, 0, N){
        ll y = (1ll << i);
        if(a[i] / x >= y){
            ans += f(y, i);
        }
    }
    return ans;
}

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