Submission #631001

#	Time	Username	Problem	Language	Result	Execution time	Memory
631001		garam1732	Rarest Insects (IOI22_insects)	C++17	64.14 / 100	170 ms	516 KiB

insects

#include "insects.h"
#include <iostream>
#include <vector>
#include <stack>
#include <set>
#include <algorithm>
#define ff first
#define ss second
using namespace std;
typedef pair<int, int> pi;

//vector<int> ans;
//stack<pi> S, v;
vector<int> v, w;
set<int> s;

int min_cardinality(int N) {
    int num = 0;
    for(int i = 0; i < N; i++) {
        move_inside(i);
        num++;
        if(press_button() > 1) {
            move_outside(i);
            num--;
            s.insert(i);
        }
    }

    int sum = num;
    int lb = 1, ub = N, mid;
    while(lb < ub) {
        mid = (lb*4+ub+4)/5;

        for(int i : s) {
            move_inside(i);
            if(press_button() > mid) {
                move_outside(i);
                v.push_back(i);
            }
            else w.push_back(i);
        }

        if(w.size()+sum == mid*num) {
            lb = mid;
            for(int t : w) s.erase(t);
            sum += w.size();
        }
        else {
            ub = mid-1;
            for(int t : v) s.erase(t);
            for(int t : w) move_outside(t);
        }
        v.clear();
        w.clear();
    }

    return lb;
}
/*
6
5 8 9 5 9 9

1
7
*/

Compilation message (stderr)

insects.cpp: In function 'int min_cardinality(int)':
insects.cpp:43:25: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   43 |         if(w.size()+sum == mid*num) {
      |            ~~~~~~~~~~~~~^~~~~~~~~~

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