Submission #307762

#	Time	Username	Problem	Language	Result	Execution time	Memory
307762		CodePlatina	Counting Mushrooms (IOI20_mushrooms)	C++14	56.93 / 100	12 ms	492 KiB

mushrooms

#include "mushrooms.h"
#include <vector>

using namespace std;

int count_mushrooms(int n)
{
    if(n <= 200)
    {
        int ret = 1;
        for(int i = 1; i < n; i += 2)
        {
            if(i == n - 1) ret += 1 - use_machine({0, i});
            else ret += 2 - use_machine({i, 0, i + 1});
        }
        return ret;
    }
    else
    {
        int ret = 1;
        vector<int> A, B;
        A.push_back(0);
        for(int i = 1; i <= 200; ++i)
        {
            int t = use_machine({0, i});
            if(t == 0) A.push_back(i), ++ret;
            else B.push_back(i);
        }
        if(A.size() >= B.size())
        {
            for(int i = 201; i < n; i += (int)A.size())
            {
                int cnt = 0;
                vector<int> tmp;
                for(int j = i; j < n && j < i + (int)A.size(); ++j)
                {
                    ++cnt;
                    tmp.push_back(A[j - i]);
                    tmp.push_back(j);
                }
                int t = use_machine(tmp);
                if(t & 1) ++t;
                t /= 2;
                ret += cnt - t;
            }
        }
        else
        {
            swap(A, B);
            for(int i = 201; i < n; i += (int)A.size())
            {
                int cnt = 0;
                vector<int> tmp;
                for(int j = i; j < n && j < i + (int)A.size(); ++j)
                {
                    ++cnt;
                    tmp.push_back(A[j - i]);
                    tmp.push_back(j);
                }
                int t = use_machine(tmp);
                if(t & 1) ++t;
                t /= 2;
                ret += t;
            }
        }
        return ret;
    }
}

• Subtask 1: 56.927 / 100