Submission #1009246

#	Time	Username	Problem	Language	Result	Execution time	Memory
1009246		oyber	Rarest Insects (IOI22_insects)	C++17	50 / 100	133 ms	696 KiB

insects

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

vector<int> uniqu;
vector<int> insects;
int N;
vector<bool> used;
vector<int> last_used;

void empty() {
  for (int i = 0; i < int(last_used.size()); i++) {
    int a = last_used[i];
    move_outside(a);
    used[a] = false;
  }
}

bool check_num(int num) {
  last_used = vector<int>(0);
  //printf("num: %d\n", num);
  for (int i = 0; i < int(insects.size()); i++) {
    if (used[insects[i]]) continue;
    //printf("insect: %d\n", insects[i]);
    move_inside(insects[i]);
    if (press_button() >= num) {
      move_outside(insects[i]);
      continue;
    }
    used[insects[i]] = true;
    last_used.push_back(insects[i]);
  }

  //printf("n1: %d\n", press_button());
  bool valid = true;
  for (int i = 0; i < int(uniqu.size()); i++) {
    //printf("n2: %d\n", press_button());
    move_inside(uniqu[i]);
    //printf("n3: %d\n", press_button());
    if (press_button() != num) {
      //printf("not valid i=%d\n", uniqu[i]);
      move_outside(uniqu[i]);
      valid = false;
      break;
    }
    move_outside(uniqu[i]);
  }

  //printf("valid: %d\n", int(valid));
  return valid;
}

int min_cardinality(int n) {
  N = n;
  used.resize(N);
  int types = 0;
  for (int i = 0; i < N; i++) {
    move_inside(i);
    if (press_button() > 1) {
      move_outside(i);
      insects.push_back(i);
      continue;
    }
    types++;
    uniqu.push_back(i);
  }

  for (int i = 0; i < int(uniqu.size()); i++) {
    move_outside(uniqu[i]);
  }


  int l = 1;
  int r = N/types+1;
  while (l < r) {
    int m = (l+r)/2;
    if (check_num(m)) {
      l = m+1;
    } else {
      r = m;
      empty();
    }
  }
  return r-1;

  /*for (int i = 1; i <= N/types; i++) {
    if (!check_num(i)) {
      return i-1;
    }
  }
  return N/types;*/
}

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