Submission #1052458

#	Time	Username	Problem	Language	Result	Execution time	Memory
1052458		mychecksedad	Packing Biscuits (IOI20_biscuits)	C++17	33 / 100	1047 ms	419176 KiB

biscuits

#include "biscuits.h"
#include<bits/stdc++.h>
using namespace std;
#define pb push_back
#define ll long long int
#define all(x) x.begin(),x.end()
#define vi vector<int>
#define pii pair<int,int>
#define ff first
#define ss second

long long count_tastiness(long long x, std::vector<long long> a) {
	if(x > 1000000) return 0;
	a.resize(63);
	vector<vector<ll>> dp(63, vector<ll>(x + 1));
	for(int i = 0; i + 1 < a.size(); ++i){
		if(a[i] > x){
			ll dif = (a[i] - x) / 2;
			a[i + 1] += dif;
			a[i] -= dif * 2;
		}
	}

	ll k = a.size();
	if(x == 1){
		vector<vector<int>> v;
		for(int i = 0; i < k; ++i){
			if(a[i] >= 1){
				if(i == 0 || a[i - 1] == 0) v.pb({a[i]});
				else v.back().pb(a[i]);
			}
		}
		ll ans = 1;
		for(auto u: v){
			ll x = 1;
			for(int i = u.size() - 1; i >= 0; --i){
				x *= 2;
				if(u[i] == 2) x += 1;
			}
			ans *= x;
		}
		return ans;
	}
	dp[0][0] = 1;
	for(int i = 0; i < k - 1; ++i){
		for(int j = 0; j <= x; ++j){
			dp[i + 1][(j + a[i]) >> 1] += dp[i][j];
			if(j + a[i] >= x)
				dp[i + 1][(j + a[i] - x) >> 1] += dp[i][j];
			// cout << dp[i][j] << ' ';
		}
		// cout << '\n';
	}
	ll ans = 0;
	for(int j = 0; j <= x; ++j) ans += dp[k - 1][j];

	return ans;
}

Compilation message (stderr)

biscuits.cpp: In function 'long long int count_tastiness(long long int, std::vector<long long int>)':
biscuits.cpp:16:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   16 |  for(int i = 0; i + 1 < a.size(); ++i){
      |                 ~~~~~~^~~~~~~~~~
biscuits.cpp:29:44: warning: narrowing conversion of 'a.std::vector<long long int>::operator[](((std::vector<long long int>::size_type)i))' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
   29 |     if(i == 0 || a[i - 1] == 0) v.pb({a[i]});
      |                                            ^
biscuits.cpp:29:44: warning: narrowing conversion of 'a.std::vector<long long int>::operator[](((std::vector<long long int>::size_type)i))' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]

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