Submission #817423

#TimeUsernameProblemLanguageResultExecution timeMemory
817423math_rabbit_1028Counting Mushrooms (IOI20_mushrooms)C++14
53.81 / 100
11 ms316 KiB
#include "mushrooms.h"
#include <bits/stdc++.h>
using namespace std;

int count_mushrooms(int n) {
    int b = max((int)sqrt(n), 2);
    vector<int> Alist, Blist;
    Alist.push_back(0);

    int i;
    for (i = 1; i < n; i++) {
        if (use_machine({i, Alist[0]}) == 0) Alist.push_back(i);
        else Blist.push_back(i);
        if (Alist.size() == b || Blist.size() == b) break;
    }
    i++;
    int ans = Alist.size();
    for (; i < n; i += b) {
        vector<int> seq;
        if (Alist.size() == b) {
            for (int j = i; j < min(i + b, n); j++) {
                seq.push_back(Alist[j - i]);
                seq.push_back(j);
            }
            ans += (min(i + b, n) - i) - (use_machine(seq) + 1) / 2;
        }
        else {
            for (int j = i; j < min(i + b, n); j++) {
                seq.push_back(Blist[j - i]);
                seq.push_back(j);
            }
            ans += (use_machine(seq) + 1) / 2;
        }
    }
    return ans;
}

Compilation message (stderr)

mushrooms.cpp: In function 'int count_mushrooms(int)':
mushrooms.cpp:14:26: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   14 |         if (Alist.size() == b || Blist.size() == b) break;
      |             ~~~~~~~~~~~~~^~~~
mushrooms.cpp:14:47: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   14 |         if (Alist.size() == b || Blist.size() == b) break;
      |                                  ~~~~~~~~~~~~~^~~~
mushrooms.cpp:20:26: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   20 |         if (Alist.size() == b) {
      |             ~~~~~~~~~~~~~^~~~
#Verdict Execution timeMemoryGrader output
Fetching results...