Submission #627917

#	Time	Username	Problem	Language	Result	Execution time	Memory
627917		tutis	Rarest Insects (IOI22_insects)	C++17	53.16 / 100	476 ms	31732 KiB

insects

#include "insects.h"
#include <bits/stdc++.h>
using namespace std;
int min_cardinality(vector<int> x)
{
    int N = x.size();
    int add2 = 0;
    int add1 = 0;
    vector<int>y;
    int lo, hi;
    auto cnt = [&](int c)->int
    {
        vector<int>v;
        y = {};
        int k = 0;
        for (int i : x)
        {
            v.push_back(i);
            move_inside(i);
            k = press_button();
            if (k + add1 == c + 1)
            {
                v.pop_back();
                move_outside(i);
                y.push_back(i);
            }
        }
        int r = v.size();
        for (int i : v)
            move_outside(i);
        return add2 + r;
    };
    int c = cnt(1);
    if (c * 2 > N)
        return 1;
    {
        x = y;
        add2 = c;
        add1 = 1;
    }
    lo = 1;
    hi = N / c;
    double bias = 3.2;
    vector<vector<pair<int, int>>>dp(hi + 1, vector<pair<int, int>>(hi + 1));
    for (int sz = 1; sz <= hi; sz++)
    {
        for (int l = 1; l + sz - 1 <= hi; l++)
        {
            int r = l + sz - 1;
            if (sz == 1)
                dp[l][r] = {0, l};
            else
            {
                int m0 = l + 1;
                int m1 = r;
                int m = (l + r + 1) / 2;
                auto f = [&](int x)
                {
                    x = max(x, l + 1);
                    x = min(x, r);
                    return max(dp[l][x - 1].first, dp[x][r].first);
                };
                for (int x : {dp[l + 1][r].second, dp[l][r - 1].second})
                {
                    for (int dx = -5; dx <= 5; dx++)
                    {
                        if (f(x + dx) < f(m))
                            m = x + dx;
                    }
                }
                dp[l][r].first = f(m) + N - lo * c;
                dp[l][r].second = m;
            }
        }
    }
    while (lo < hi)
    {
        int m = dp[lo][hi].second;
        m = max(m, lo + 1);
        m = min(m, hi);
        int k = cnt(m);
        if (k == m * c)
        {
            lo = m;
            x = y;
            add2 = m * c;
            add1 = m;
        }
        else
            hi = m - 1;
    }
    return lo;
}
int min_cardinality(int N) {
    vector<int>v;
    for (int i = 0; i < N; i++)
        v.push_back(i);
    return min_cardinality(v);
}

Compilation message (stderr)

insects.cpp: In function 'int min_cardinality(std::vector<int>)':
insects.cpp:54:21: warning: unused variable 'm0' [-Wunused-variable]
   54 |                 int m0 = l + 1;
      |                     ^~
insects.cpp:55:21: warning: unused variable 'm1' [-Wunused-variable]
   55 |                 int m1 = r;
      |                     ^~
insects.cpp:43:12: warning: unused variable 'bias' [-Wunused-variable]
   43 |     double bias = 3.2;
      |            ^~~~

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