Submission #566798

#	Time	Username	Problem	Language	Result	Execution time	Memory
566798		2fat2code	Packing Biscuits (IOI20_biscuits)	C++17	21 / 100	1079 ms	340 KiB

biscuits

#include "biscuits.h"
#include <bits/stdc++.h>
#define fr first
#define sc second
#define all(s) s.begin(), s.end()
#define int long long
using namespace std;

int ans, k;

int count_tastiness(int x, vector<int> a) {
    ans = 1LL;
    if(x == 1){
        k = (int)a.size();
        for(int i=0;i<(int)a.size();i++){
            if(a[i] >= 3){
                if(i == a.size() - 1){
                    a.push_back(0);
                }
                a[i + 1] += ( (a[i] - 1LL) / 2LL);
                if(a[i] % 2 == 1LL) a[i] = 1;
                else a[i]= 2;
            }
        }
        int j = 0;
        while(j < (int)a.size()){
            int curr = 0;
            int pos = j;
            while(j < (int)a.size() && a[j] != 0){
                curr = (curr + a[j] * (1LL << (j - pos)));
                ++j;
            }
            curr++;
            ans *= curr;
            j++;
        }
    }
    else{
        ans = 1;
	int sum = 0;
	int nrbiti = (int)a.size();
	for(int i=0;i<nrbiti;i++){
        sum += a[i] * (1LL << i);
	}
        for(int curr=1;curr<=(sum/x);curr++){
        vector<int>b = a;
        b.push_back(0);
        for(int i=0;(1LL<<i)<=curr;i++){
            if(i == b.size() - 1) b.push_back(0);
            if(curr & (1LL<<i) && b[i] < x){
                goto next;
            }
            else{
                if(curr & (1LL << i)) b[i] -= x;
                b[i+1] += (b[i] / 2);
            }
        }
        ++ans;
        next:;
        }
    }
	return ans;
}

Compilation message (stderr)

biscuits.cpp: In function 'long long int count_tastiness(long long int, std::vector<long long int>)':
biscuits.cpp:17:22: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   17 |                 if(i == a.size() - 1){
      |                    ~~^~~~~~~~~~~~~~~
biscuits.cpp:49:18: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   49 |             if(i == b.size() - 1) b.push_back(0);
      |                ~~^~~~~~~~~~~~~~~

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