Submission #721520

#TimeUsernameProblemLanguageResultExecution timeMemory
721520vjudge1Counting Mushrooms (IOI20_mushrooms)C++17
Compilation error
0 ms0 KiB
#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 mn = 10000, ans;
    for(int i = 1; i < 500; i++){
        if(2*i + ((n - (2*i)) / i) < mn){
            mn = 2*i + ((n - (2*i)) / i);
            ans = i;
        }
    }

    int unt = ans*2;
    // 188 - 1 = 187
    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;
    vector <int> Is;
    // n - 188 / 94
    if(As.size() >= Bs.size()){
        for(; i < n;){
            set <int> search;
            int j = 0;
            for(; j < As.size() && i < n; j++ ){
                search.push_back(i);
                Is.push_back(i);
                search.push_back(As[j]);
                i++;
            }
            int q = use_machine(search);
            cnt +=  (search.size()/2) - ((q/2) +(q % 2));
            if(q/2 + q % 2 == 0){
                for(int ii = 0; ii < Is.size(); ii++){
                  As.push_back(Is[ii]);
                }
            }
        }
    }
    else{
        for(; i < n; ){
            set <int> search;
            for(int j = 0; j < Bs.size() && i < n; j++){
                search.push_back(i);
                Is.push_back(i);
                search.push_back(Bs[j]);
                i++;
            }

            int q = use_machine(search);
            cnt +=  q/2 + q % 2;
            if(q/2 + q % 2 == 0){
                for(int ii = 0; ii < Is.size(); ii++){
                  Bs.push_back(Is[ii]);
                }
            }
        }   
    }
    // 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:48:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   48 |             for(; j < As.size() && i < n; j++ ){
      |                   ~~^~~~~~~~~~~
mushrooms.cpp:49:24: error: 'class std::set<int>' has no member named 'push_back'
   49 |                 search.push_back(i);
      |                        ^~~~~~~~~
mushrooms.cpp:51:24: error: 'class std::set<int>' has no member named 'push_back'
   51 |                 search.push_back(As[j]);
      |                        ^~~~~~~~~
mushrooms.cpp:54:33: error: could not convert 'search' from 'std::set<int>' to 'std::vector<int>'
   54 |             int q = use_machine(search);
      |                                 ^~~~~~
      |                                 |
      |                                 std::set<int>
mushrooms.cpp:57:36: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   57 |                 for(int ii = 0; ii < Is.size(); ii++){
      |                                 ~~~^~~~~~~~~~~
mushrooms.cpp:66:30: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   66 |             for(int j = 0; j < Bs.size() && i < n; j++){
      |                            ~~^~~~~~~~~~~
mushrooms.cpp:67:24: error: 'class std::set<int>' has no member named 'push_back'
   67 |                 search.push_back(i);
      |                        ^~~~~~~~~
mushrooms.cpp:69:24: error: 'class std::set<int>' has no member named 'push_back'
   69 |                 search.push_back(Bs[j]);
      |                        ^~~~~~~~~
mushrooms.cpp:73:33: error: could not convert 'search' from 'std::set<int>' to 'std::vector<int>'
   73 |             int q = use_machine(search);
      |                                 ^~~~~~
      |                                 |
      |                                 std::set<int>
mushrooms.cpp:76:36: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   76 |                 for(int ii = 0; ii < Is.size(); ii++){
      |                                 ~~~^~~~~~~~~~~