Submission #490291

#TimeUsernameProblemLanguageResultExecution timeMemory
490291mraronPacking Biscuits (IOI20_biscuits)C++14
42 / 100
1092 ms49860 KiB
#include "biscuits.h"
#include <iostream>
#include <unordered_map>
#include <map>
#include <utility>
#include <functional>
#include <cassert>
using namespace std;

using ll = long long ;

ll x;
vector<ll> a;


unordered_map<ll, ll> dp[61];

ll calc(ll rem, ll i) {
	if(dp[i].count(rem)) return dp[i][rem];
	ll orig=rem;
	
	if(i<(int)a.size() || rem>=x) {
		if(i<(int)a.size()) rem+=a[i];
		
		ll ans=calc(rem/2, i+1);
		if(rem>=x) {
			ans+=calc((rem-x)/2, i+1);
		}
		
		return dp[i][orig]=ans;
	}else return 1;
}	

ll count_tastiness(ll x_, vector<ll> a_) {
	a=a_;
	x=x_;

	a.resize(60);
	for(int i=0;i<60;++i) {
		if(a[i]>=x+2) {
			ll diff=a[i]-(x+1);
			a[i]-=(diff+1)/2*2;
			a[i+1]+=(diff+1)/2;
		}
	}
	
	for(auto& i:dp) i.clear();
	
	ll res=calc(0, 0);
	
	return res;
}

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