제출 #1223873

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1223873		bangan	드문 곤충 (IOI22_insects)	C++20	57.14 / 100	41 ms	448 KiB

insects

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

#define pb push_back

mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count());

int n;
int unq;
vector<int> in;

bool ok(int k) {
    while (!in.empty()) {
        move_outside(in.back());
        in.pop_back();
    }
    vector<int> all;
    for (int i=0; i<n; i++) all.pb(i);
    shuffle(all.begin(), all.end(), rng);
    for (int i : all) {
        move_inside(i);
        if (press_button() > k) move_outside(i);
        else in.pb(i);
        if (unq * k == in.size()) return true;
    }
    return false;
}

int min_cardinality(int N) {
    for (int i = 0; i < N; i++) {
        move_inside(i);
        if (press_button() == 2) move_outside(i);
        else in.pb(i);
    }

    unq = in.size();
    while (!in.empty()) {
        move_outside(in.back());
        in.pop_back();
    }

    if (unq <= 6) {
        vector<int> all;
        for (int i=0; i<N; i++) all.pb(i);

        unq--;
        int ans = N;
        while (unq--) {
            vector<int> rem;
            int cur = 0;
            while (!all.empty()) {
                int i = all.back();
                all.pop_back();
                move_inside(i);
                int res = press_button();
                if (res == cur) {
                    move_outside(i);
                    rem.pb(i);
                }
                else {
                    in.pb(i);
                }
                cur = res;
            }
            ans = min(ans, cur);
            all = rem;
            while (!in.empty()) {
                move_outside(in.back());
                in.pop_back();
            }
        }
        return min((int) all.size(), ans);
    }

    n = N;
    int l = 1, r = n / unq;
    while (l != r) {
        int mid = (l+r+1) / 2;
        ok(mid) ? l = mid : r = mid - 1;
    }
    return l;
}

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