Submission #1069846

#TimeUsernameProblemLanguageResultExecution timeMemory
1069846j_vdd16Rarest Insects (IOI22_insects)C++17
70.14 / 100
61 ms644 KiB
#include "insects.h"

#include <algorithm>
#include <bitset>
#include <cstdint>
#include <cstring>
#include <iostream>
#include <limits.h>
#include <math.h>
#include <map>
#include <numeric>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <vector>

//#define int long long
#define loop(X, N) for(int X = 0; X < (N); X++)
#define all(V) V.begin(), V.end()
#define rall(V) V.rbegin(), V.rend()

using namespace std;

typedef vector<int> vi;
typedef vector<vi> vvi;
typedef pair<int, int> ii;
typedef vector<ii> vii;
typedef vector<vector<ii>> vvii;
typedef vector<bool> vb;
typedef vector<vector<bool>> vvb;

typedef uint64_t u64;
typedef int64_t i64;

int min_cardinality(int N) {
    vb isDone(N);

    vb isRemoved(N);
    loop(i, N) {
        move_inside(i);

        int res = press_button();
        if (res > 1) {
            move_outside(i);
            isRemoved[i] = true;
        }
    }

    int diffCount = 0;
    loop(i, N) {
        if (!isRemoved[i]) {
            diffCount++;
            isDone[i] = true;
        }
    }

    loop(i, N) {
        if (isDone[i]) continue;

        move_inside(i);
    }
    int maxCount = press_button();
    loop(i, N) {
        if (isDone[i]) continue;

        move_outside(i);
    }

    int total = N;
    vi possible;
    for (int x = 1; x <= N; x++) {
        if (diffCount == 2 && x + maxCount != total) continue;
        if (diffCount == 1 && (x != maxCount || x != total)) continue;

        if ((diffCount - 1) * x + maxCount <= total && total <= (diffCount - 1) * maxCount + x) {
            possible.push_back(x);
        }
    }

    vb madeItBigger(N, false);
    int extra = 1;
    int l = 0, r = possible.size() - 1;
    while (l < r) {
        int m = (3 * l + r + 3) / 4;
        //int m = (l + r + 1) / 2;

        int lessThanM = diffCount * extra;
        isRemoved = vb(N);
        loop(i, N) {
            if (isDone[i]) continue;

            move_inside(i);

            if (madeItBigger[i] || press_button() > possible[m]) {
                madeItBigger[i] = true;
                move_outside(i);
                isRemoved[i] = true;
            }
            else {
                lessThanM++;
            }
        }

        if (lessThanM == diffCount * possible[m]) {
            l = m;
            loop(i, N) {
                if (!isRemoved[i] && !isDone[i]) {
                    isDone[i] = true;
                }
            }

            extra = possible[m];
            madeItBigger = vb(N, false);
        }
        else {
            r = m - 1;

            loop(i, N) {
                if (!isRemoved[i] && !isDone[i]) {
                    move_outside(i);
                }
            }
        }
    }

    return possible[l];
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...