Submission #1117306

#	Time	Username	Problem	Language	Result	Execution time	Memory
1117306		pokmui9909	Counting Mushrooms (IOI20_mushrooms)	C++17	80.43 / 100	10 ms	596 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, 140), 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++;
        }
    }
    B = max(P.size(), Q.size());
    ll Rev = 0;
    if(Q.size() == B){
        Rev = 1;
        swap(P, Q);
    }
    ll Ans = P.size();
    for(ll i = S; i < N; i += B){
        vector<int> V;
        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);
    }
    return (Rev == 0 ? Ans : N - 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:52:17: 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]
   52 |     if(Q.size() == B){
      |        ~~~~~~~~~^~~~

• Subtask 1: 80.427 / 100