Submission #1117321

#	Time	Username	Problem	Language	Result	Execution time	Memory
1117321		pokmui9909	Counting Mushrooms (IOI20_mushrooms)	C++17	90.40 / 100	9 ms	856 KiB

mushrooms

#include "mushrooms.h"
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
 
#define x first
#define y second
 
int count_mushrooms(int N){
    ll B = min(N, 120), S = 1;
    vector<ll> P, Q;
    P.push_back(0);
    while(S < N && P.size() < B && Q.size() < B){
        if(S + 1 < N && P.size() >= 2){
            ll f = use_machine((vector<int>){P[0], S, P[1], S + 1});
            if(f / 2 == 0){
                P.push_back(S);
            } else {
                Q.push_back(S);
            }
            if(f % 2 == 0){
                P.push_back(S + 1);
            } else {
                Q.push_back(S + 1);
            }
            S += 2;
        } else if(S + 1 < N && Q.size() >= 2){
            ll f = use_machine((vector<int>){Q[0], S, Q[1], S + 1});
            if(f / 2 == 1){
                P.push_back(S);
            } else {
                Q.push_back(S);
            }
            if(f % 2 == 1){
                P.push_back(S + 1);
            } else {
                Q.push_back(S + 1);
            }
            S += 2;
        } else {
            ll f = use_machine((vector<int>){0, S});
            if(f == 0){
                P.push_back(S);
            } else {
                Q.push_back(S);
            }
            S++;
        }
    }
    ll Ans = P.size();
    for(ll i = S; i < N; i += B){
        B = max(P.size(), Q.size());
        vector<int> V;
        if(P.size() == B){
            for(ll j = 0; j < B; j++){
                if(i + j >= N) break;
                V.push_back(P[j]);
                V.push_back(i + j);
                Ans++;
            }
            ll t = use_machine(V);
            Ans -= (t / 2) + (t % 2);
            if(t % 2 == 0){
                P.push_back(V.back());
            } else {
                Q.push_back(V.back());
            }
        } else {
            for(ll j = 0; j < B; j++){
                if(i + j >= N) break;
                V.push_back(Q[j]);
                V.push_back(i + j);
            }
            ll t = use_machine(V);
            Ans += (t / 2) + (t % 2);
            if(t % 2 == 1){
                P.push_back(V.back());
            } else {
                Q.push_back(V.back());
            }
        }
    }
    return Ans;
}

Compilation message (stderr)

mushrooms.cpp: In function 'int count_mushrooms(int)':
mushrooms.cpp:13:29: warning: comparison of integer expressions of different signedness: 'std::vector<long long int>::size_type' {aka 'long unsigned int'} and 'll' {aka 'long long int'} [-Wsign-compare]
   13 |     while(S < N && P.size() < B && Q.size() < B){
      |                    ~~~~~~~~~^~~
mushrooms.cpp:13:45: warning: comparison of integer expressions of different signedness: 'std::vector<long long int>::size_type' {aka 'long unsigned int'} and 'll' {aka 'long long int'} [-Wsign-compare]
   13 |     while(S < N && P.size() < B && Q.size() < B){
      |                                    ~~~~~~~~~^~~
mushrooms.cpp:15:66: warning: narrowing conversion of 'P.std::vector<long long int>::operator[](0)' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
   15 |             ll f = use_machine((vector<int>){P[0], S, P[1], S + 1});
      |                                                                  ^
mushrooms.cpp:15:66: warning: narrowing conversion of 'P.std::vector<long long int>::operator[](0)' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
mushrooms.cpp:15:52: warning: narrowing conversion of 'S' from 'll' {aka 'long long int'} to 'int' [-Wnarrowing]
   15 |             ll f = use_machine((vector<int>){P[0], S, P[1], S + 1});
      |                                                    ^
mushrooms.cpp:15:52: warning: narrowing conversion of 'S' from 'll' {aka 'long long int'} to 'int' [-Wnarrowing]
mushrooms.cpp:15:66: warning: narrowing conversion of 'P.std::vector<long long int>::operator[](1)' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
   15 |             ll f = use_machine((vector<int>){P[0], S, P[1], S + 1});
      |                                                                  ^
mushrooms.cpp:15:66: warning: narrowing conversion of 'P.std::vector<long long int>::operator[](1)' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
mushrooms.cpp:15:63: warning: narrowing conversion of '(S + 1)' from 'll' {aka 'long long int'} to 'int' [-Wnarrowing]
   15 |             ll f = use_machine((vector<int>){P[0], S, P[1], S + 1});
      |                                                             ~~^~~
mushrooms.cpp:15:63: warning: narrowing conversion of '(S + 1)' from 'll' {aka 'long long int'} to 'int' [-Wnarrowing]
mushrooms.cpp:28:66: warning: narrowing conversion of 'Q.std::vector<long long int>::operator[](0)' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
   28 |             ll f = use_machine((vector<int>){Q[0], S, Q[1], S + 1});
      |                                                                  ^
mushrooms.cpp:28:66: warning: narrowing conversion of 'Q.std::vector<long long int>::operator[](0)' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
mushrooms.cpp:28:52: warning: narrowing conversion of 'S' from 'll' {aka 'long long int'} to 'int' [-Wnarrowing]
   28 |             ll f = use_machine((vector<int>){Q[0], S, Q[1], S + 1});
      |                                                    ^
mushrooms.cpp:28:52: warning: narrowing conversion of 'S' from 'll' {aka 'long long int'} to 'int' [-Wnarrowing]
mushrooms.cpp:28:66: warning: narrowing conversion of 'Q.std::vector<long long int>::operator[](1)' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
   28 |             ll f = use_machine((vector<int>){Q[0], S, Q[1], S + 1});
      |                                                                  ^
mushrooms.cpp:28:66: warning: narrowing conversion of 'Q.std::vector<long long int>::operator[](1)' from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' [-Wnarrowing]
mushrooms.cpp:28:63: warning: narrowing conversion of '(S + 1)' from 'll' {aka 'long long int'} to 'int' [-Wnarrowing]
   28 |             ll f = use_machine((vector<int>){Q[0], S, Q[1], S + 1});
      |                                                             ~~^~~
mushrooms.cpp:28:63: warning: narrowing conversion of '(S + 1)' from 'll' {aka 'long long int'} to 'int' [-Wnarrowing]
mushrooms.cpp:41:49: warning: narrowing conversion of 'S' from 'll' {aka 'long long int'} to 'int' [-Wnarrowing]
   41 |             ll f = use_machine((vector<int>){0, S});
      |                                                 ^
mushrooms.cpp:41:49: warning: narrowing conversion of 'S' from 'll' {aka 'long long int'} to 'int' [-Wnarrowing]
mushrooms.cpp:54:21: warning: comparison of integer expressions of different signedness: 'std::vector<long long int>::size_type' {aka 'long unsigned int'} and 'll' {aka 'long long int'} [-Wsign-compare]
   54 |         if(P.size() == B){
      |            ~~~~~~~~~^~~~

• Subtask 1: 90.4 / 100