Submission #721489

#	Time	Username	Problem	Language	Result	Execution time	Memory
721489		vjudge1	Counting Mushrooms (IOI20_mushrooms)	C++17	26.68 / 100	17 ms	208 KiB

mushrooms

#include "mushrooms.h"
#include <bits/stdc++.h>
using namespace std;

#define N '\n';
#define VI <vector <int>>
#define VVI <vector <vector <int>>>  

int count_mushrooms(int n){
    int cnt = 1;    
    
    if(n <= 226){
        for(int i = 1; i < n; i++){
            cnt += 1 - use_machine({0, i});
        }
        return cnt;
    }

    vector <int> As = {0};
    vector <int> Bs;
    int unt = 50;
    for(int i = 1; i < unt; i++){
        bool q = use_machine({0,i});
        if(q)
            Bs.push_back(i);
        else
            As.push_back(i);
    }
    cnt += As.size()-1;
    
    int i = unt;
    if(As.size() >= Bs.size()){
        for(; i < n;){
            vector<int> search;
            int j = 0;
            for(; j < As.size() && i < n; j++ ){
                search.push_back(i);
                search.push_back(As[j]);
                i++;
            }
            int q = use_machine(search);
            cnt +=  (search.size()/2) - ((q/2) +(q % 2));
        }
    }
    else{
        for(; i < n; ){
            vector<int> search;
            for(int j = 0; j < Bs.size() && i < n; j++){
                search.push_back(i);
                search.push_back(Bs[j]);
                i++;
            }

            int q = use_machine(search);
            cnt +=  q/2 + q % 2;
        }   
    }
    // B A B A B A
    // B A A A B A B A
    return cnt;
}

Compilation message (stderr)

mushrooms.cpp: In function 'int count_mushrooms(int)':
mushrooms.cpp:36:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   36 |             for(; j < As.size() && i < n; j++ ){
      |                   ~~^~~~~~~~~~~
mushrooms.cpp:48:30: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   48 |             for(int j = 0; j < Bs.size() && i < n; j++){
      |                            ~~^~~~~~~~~~~

• Subtask 1: 26.6824 / 100