Submission #425601

#	Time	Username	Problem	Language	Result	Execution time	Memory
425601		idk321	Packing Biscuits (IOI20_biscuits)	C++17	9 / 100	223 ms	332 KiB

biscuits

#include "biscuits.h"
#include <bits/stdc++.h>;
using namespace std;
typedef long long ll;


long long count_tastiness(long long x, std::vector<long long> a) {
    ll sum = 0;
    int k = 60;
    for (int i = 0; i < a.size(); i++)
    {
        sum += (1LL << i) * a[i];
    }




    if (sum <= 100000)
    {
        int res = 0;

        for (int i = 0; i <= sum; i++)
        {
            bool poss = true;
            vector<ll> cur(60);
            for (int j = 0; j < a.size(); j++) cur[j] = a[j];
            for (int j = 0; j < k; j++)
            {
                if ((1LL << j) & i)
                {
                    cur[j] -= x;

                    if (cur[j] < 0)
                    {
                        poss = false;
                        break;
                    }
                }


                if (j + 1 < k)
                {
                    cur[j + 1] += cur[j] / 2;
                }
            }


            if (poss) res++;
        }

        return res;
    }

    if (x == 1)
    {
        vector<ll> cur(k);
        for (int j = 0; j < a.size(); j++) cur[j] = a[j];
        for (int j = 0; j < k - 1; j++)
        {
            if (cur[j] == 0) continue;
            if (cur[j] % 2 == 0)
            {
                cur[j + 1] += (cur[j] - 2) / 2;
                cur[j] = 2;
            }
            if (cur[j] % 2 == 1)
            {
                cur[j + 1] += cur[j] / 2;
                cur[j] = 1;
            }
        }

        ll res = 1;
        for (int i = 0; i < k; i++)
        {
            for (int j = 0; j < cur[i]; j++)
            {
                res *= 2;
            }
        }

        return res;
    }

	return 0;
}

/*
1
3 3
5 2 1
*/

Compilation message (stderr)

biscuits.cpp:2:25: warning: extra tokens at end of #include directive
    2 | #include <bits/stdc++.h>;
      |                         ^
biscuits.cpp: In function 'long long int count_tastiness(long long int, std::vector<long long int>)':
biscuits.cpp:10:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   10 |     for (int i = 0; i < a.size(); i++)
      |                     ~~^~~~~~~~~~
biscuits.cpp:26:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   26 |             for (int j = 0; j < a.size(); j++) cur[j] = a[j];
      |                             ~~^~~~~~~~~~
biscuits.cpp:57:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   57 |         for (int j = 0; j < a.size(); j++) cur[j] = a[j];
      |                         ~~^~~~~~~~~~

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