제출 #824387

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
824387		ymm	비스킷 담기 (IOI20_biscuits)	C++17	100 / 100	59 ms	1236 KiB

biscuits

#include "biscuits.h"
#include <bits/stdc++.h>
#define Loop(x,l,r) for (ll x = (l); x < (r); ++x)
typedef long long ll;
typedef std::pair<ll,ll> pll;
using namespace std;

const int N = 64;
ll a[N], b[N];
map<pll,ll> mem;
int n;

ll solve(int k, ll mx)
{
	mx = min(mx, (1ll<<k)-1);
	if (k == 0 || mx == 0)
		return 1;
	auto &ans = mem[pll(k, mx)];
	if (ans)
		return ans;
	ans += solve(k-1, min(mx, b[k-1]));
	if (mx >= (1ll << (k-1)))
		ans += solve(k-1, min(mx - (1ll << (k-1)), b[k-1]));
	return ans;
}

long long count_tastiness(long long x, std::vector<long long> _a) {
	_a.resize(61, 0);
	n = _a.size();
	Loop (i,0,n)
		a[i] = _a[i];
	ll sum = 0;
	Loop (i,0,n) {
		sum += a[i]<<i;
		b[i+1] = sum/x;
	}
	mem.clear();
	return solve(n, b[n]);
}

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