Submission #805192

#	Time	Username	Problem	Language	Result	Execution time	Memory
805192		alvingogo	Packing Biscuits (IOI20_biscuits)	C++14	100 / 100	12 ms	1324 KiB

biscuits

#include "biscuits.h"
#include <bits/stdc++.h>
#pragma GCC optimize("Ofast")
#define fs first
#define sc second
#define p_q priority_queue
using namespace std;
typedef long long ll;

ll count_tastiness(ll x, vector<ll> a) {
	{
		vector<ll> c(a.size()+1);
		for(int i=0;i<a.size();i++){
			c[i+1]=a[i];
		}
		a=c;
	}
	while(a.size()<=63){
		a.push_back(0);
	}
	vector<ll> b=a;
	int n=a.size();
	for(int i=1;i<n;i++){
		b[i]+=b[i-1]/2;
	}
	vector<ll> dp(n),pre(n),cost(n),cp(n);
	dp[0]=1;
	pre[0]=1;
	cost[0]=9e18;
	for(int i=1;i<n;i++){
		if(b[i]<x){
			dp[i]=0;
			pre[i]=pre[i-1];
			cost[i]=-1;
			cp[0]=9e18;
			continue;
		}
		if(a[i]>=x){
			dp[i]=pre[i-1];
			pre[i]=2*dp[i];
			cost[i]=b[i]-x;
			cp[i]=0;
			continue;
		}
		ll f=x-a[i];
		for(int j=i-1;j>0;j--){
			f*=2;
			if(cost[j]>=0){
				if(cost[j]>=f){
					f-=a[j]-x;
					f=max(f,cp[j]);
					if(j){
						dp[i]+=pre[j-1];
					}
				}
				else{
					f-=a[j];
				}
			}
			else{
				f-=a[j];
			}
		}
		dp[i]++;
		pre[i]=pre[i-1]+dp[i];
		cost[i]=b[i]-x;
		cp[i]=x-a[i];
	}
	return pre[n-1];
}

Compilation message (stderr)

biscuits.cpp: In function 'll count_tastiness(ll, std::vector<long long int>)':
biscuits.cpp:13:16: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   13 |   for(int i=0;i<a.size();i++){
      |               ~^~~~~~~~~

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