Submission #629599

#	Time	Username	Problem	Language	Result	Execution time	Memory
629599		flashmt	Rarest Insects (IOI22_insects)	C++17	0 / 100	5 ms	304 KiB

insects

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

// return (found max cardinality, good IDs, bad IDs)
tuple<int, vector<int>, vector<int>> findFirstOccurrences(vector<int> id, int maxC, int expectedSize)
{
  vector<int> good, bad;
  int foundMaxC = 0;
  for (int i : id)
  {
    if (size(good) == expectedSize)
    {
      bad.push_back(i);
      continue;
    }

    move_inside(i);
    foundMaxC = max(foundMaxC, press_button());
    if (foundMaxC <= maxC) good.push_back(i);
    else
    {
      move_outside(i);
      bad.push_back(i);
    }
  }
  for (int i : good)
    move_outside(i);
  return {foundMaxC, good, bad};
}

int min_cardinality(int n)
{
  vector<int> id(n);
  iota(begin(id), end(id), 0);
  auto [foundMaxC, good, bad] = findFirstOccurrences(id, 1, n);
  int typeCnt = size(good);

  int low = 2, high = n / typeCnt, ans = 1;
  id = bad;
  while (low <= high)
  {
    int mid = (low + high + 1) / 2;
    int expectedSize = typeCnt * (mid - ans);
    tie(foundMaxC, good, bad) = findFirstOccurrences(id, mid - ans, expectedSize);
    if (size(good) == expectedSize)
    {
      ans = mid;
      low = mid + 1;
      id = bad;
    }
    else
    {
      high = min(mid - 1, min(int(size(good)) / typeCnt, foundMaxC) + ans);
      id = good;
    }
  }
  return ans;
}

Compilation message (stderr)

insects.cpp: In function 'std::tuple<int, std::vector<int, std::allocator<int> >, std::vector<int, std::allocator<int> > > findFirstOccurrences(std::vector<int>, int, int)':
insects.cpp:12:20: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   12 |     if (size(good) == expectedSize)
      |         ~~~~~~~~~~~^~~~~~~~~~~~~~~
insects.cpp: In function 'int min_cardinality(int)':
insects.cpp:46:20: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   46 |     if (size(good) == expectedSize)
      |         ~~~~~~~~~~~^~~~~~~~~~~~~~~

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