Submission #1078367

#	Time	Username	Problem	Language	Result	Execution time	Memory
1078367		allin27x	Rarest Insects (IOI22_insects)	C++17	100 / 100	41 ms	932 KiB

insects

#include <bits/stdc++.h>
#include "insects.h"
using namespace std;
#define int long long

//void move_inside(int)
//void move_outside(int)
// int press_button()


signed min_cardinality(signed N){
    std::random_device rd;
    std::mt19937 g(rd());
    vector<int> in,out; int k;
    for (int i=0; i<N; i++) {
        move_inside(i);
        if (press_button() == 2) move_outside(i);
        else in.push_back(i);
    }
    k = in.size(); for (int x: in) move_outside(x);
    int ans = 0;
    vector<int> tot; for (int i=0; i<N; i++) tot.push_back(i);
    while (1) {
        int mx =  (int)tot.size()/k;
        if (!mx) return ans;
        int m = (mx+1) / 2;
        in.clear(); out.clear();
        shuffle(tot.begin(), tot.end(), g);

        for (int x: tot) {
            if (in.size()==m*k) {out.push_back(x); continue;}
            move_inside(x);
            if (press_button() == m+1) move_outside(x), out.push_back(x);
            else in.push_back(x);
        }

        if (in.size() == m * k) {
            ans += m; tot = out;
        } else {
            tot = in;
        }
        for (int x: in) move_outside(x);
    }
}

Compilation message (stderr)

insects.cpp: In function 'int min_cardinality(int)':
insects.cpp:31:26: warning: comparison of integer expressions of different signedness: 'std::vector<long long int>::size_type' {aka 'long unsigned int'} and 'long long int' [-Wsign-compare]
   31 |             if (in.size()==m*k) {out.push_back(x); continue;}
      |                 ~~~~~~~~~^~~~~
insects.cpp:37:23: warning: comparison of integer expressions of different signedness: 'std::vector<long long int>::size_type' {aka 'long unsigned int'} and 'long long int' [-Wsign-compare]
   37 |         if (in.size() == m * k) {
      |             ~~~~~~~~~~^~~~~~~~

• Subtask 1: 10 / 10
• Subtask 2: 15 / 15
• Subtask 3: 75 / 75