Submission #721440

#	Time	Username	Problem	Language	Result	Execution time	Memory
721440		vjudge1	Counting Mushrooms (IOI20_mushrooms)	C++17	0 / 100	2 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 = 10;
    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;

    if(As.size() >= Bs.size()){
        for(int i = unt; i + As.size() < n; i+= As.size()){
            vector<int> search;
            for(int j = 0, i2 = i; j < As.size(); j++, i2++){
                search.push_back(i);
                search.push_back(As[j]);
            }
            int q = use_machine(search);
            cnt +=  As.size()- q/2 + q % 2;
        }
    }
    else{
        for(int i = unt; i + Bs.size() < n; i+= Bs.size()){
            vector<int> search;
            for(int j = 0, i2 = i; j < Bs.size(); j++, i2++){
                search.push_back(i);
                search.push_back(Bs[j]);
            }
            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:32:40: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   32 |         for(int i = unt; i + As.size() < n; i+= As.size()){
      |                          ~~~~~~~~~~~~~~^~~
mushrooms.cpp:34:38: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   34 |             for(int j = 0, i2 = i; j < As.size(); j++, i2++){
      |                                    ~~^~~~~~~~~~~
mushrooms.cpp:43:40: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   43 |         for(int i = unt; i + Bs.size() < n; i+= Bs.size()){
      |                          ~~~~~~~~~~~~~~^~~
mushrooms.cpp:45:38: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   45 |             for(int j = 0, i2 = i; j < Bs.size(); j++, i2++){
      |                                    ~~^~~~~~~~~~~

• Subtask 1: 0 / 100