Submission #490297

#TimeUsernameProblemLanguageResultExecution timeMemory
490297mraronPacking Biscuits (IOI20_biscuits)C++14
77 / 100
1089 ms1780 KiB
#include "biscuits.h"
#include <iostream>
#include <unordered_map>
#include <map>
#include <utility>
#include <functional>
#include <cassert>
#include <set>
using namespace std;

using ll = long long ;

ll x;
vector<ll> a;


map<ll, ll> dp[61];

ll calc(ll rem, ll i) {
	if(dp[i].count(rem)) return dp[i][rem];
	
	{
		auto it=dp[i].lower_bound(rem);
		if(it!=dp[i].end() && it!=dp[i].begin()) {
			if(it->second==prev(it)->second) {
				return dp[i][rem]=it->second;
			}
		}
	}
	
	ll orig=rem;
	
	if(i<(int)a.size() || rem>=x) {
		if(i<(int)a.size()) rem+=a[i];
		else return 1LL<<(rem/x);
		
		ll ans=0;
		
		if(rem>=x) {
			if(rand()&1) {
				ans+=calc(rem/2, i+1);
				ans+=calc((rem-x)/2, i+1);
			}else {
				ans+=calc((rem-x)/2, i+1);
				ans+=calc(rem/2, i+1);
			}
		}else {
			ans+=calc(rem/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;
		}
	}
	
	while(a.back()==0) a.pop_back();
	
	for(auto& i:dp) i.clear();
	
	ll res=calc(0, 0);
	
	//~ for(int i=0;i<60;++i) {
		//~ set<array<ll, 2>> volt;
		//~ for(auto j:dp[i]) {
			//~ array<ll, 2> akt={j.second, __builtin_clz(j.first)};
			//~ if(!volt.count(akt)) {
				//~ cerr<<i<<","<<j.first<<" -> "<<j.second<<" "<<__builtin_clz(j.first)<<"\n"; 
				//~ volt.insert(akt);
			//~ }
		//~ }
	//~ }
		
	std::cerr<<res<<"\n";
	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...