Submission #1235673

#TimeUsernameProblemLanguageResultExecution timeMemory
1235673Muhammad_AneeqRarest Insects (IOI22_insects)C++17
99.95 / 100
15 ms432 KiB
#include "insects.h"
#include <vector>
#include <bitset>
#include <map>
#include <set>
#include <iostream>
using namespace std;
int const sz=2010;
bitset<sz>ind,ind1,alw;
bitset<sz>ver[sz]={};
int pre;
bool chn;
int ask()
{
    if (chn==0)
        return pre;
    pre=press_button();
    chn=0;
    return pre;
}
int min_cardinality(int N) 
{
    int colors=0;
    for (int i=0;i<N;i++)
    {
        ind[i]=1;
        chn=1;
        move_inside(i);
        colors++;
        if (ask()>1)
        {
            ind[i]=0;
            chn=1;
            move_outside(i);
            colors--;
        }
        else
            alw[i]=1;
    }
    ver[1]=ind;
    set<int>s;
    s.insert(1);
    for (int i=0;i<N;i++)
        ind1[i]=1-ind[i];
    int cnt=colors;
    int st=1,en=N/colors+1;
    while (st+1<en)
    {
        int mid=(st+en)/2;
        auto z=s.lower_bound(mid);
        z--;
        int pre=*z;
        for (int i=0;i<N;i++)
        {
            if (ver[pre][i]==ind[i]) continue;
            if (ver[pre][i])
            {
                chn=1;
                move_inside(i);
                ind[i]=1;
                ind1[i]=0;
                cnt++;
            }   
            else
            {
                chn=1;
                move_outside(i);
                ind[i]=0;
                ind1[i]=1;
                cnt--;
            }
        }
        for (int i=0;i<N;i++)
        {
            if (cnt==colors*mid)
                break;
            if (ind1[i])
            {
                chn=1;
                move_inside(i);
                ind[i]=1;
                cnt++;
                if (ask()>mid)
                {
                    cnt--;
                    chn=1;
                    move_outside(i);
                    ind[i]=0;
                }
            }
        }
        if (cnt==colors*mid)
        {
            st=mid;
            if (st+1>=en)
                break;
            for (int i=0;i<N;i++)
            {
                if (ind[i])
                    ind1[i]=0;
            }
        }
        else
        {
            en=mid;
            if (st+1>=en)
                break;
            for (int i=0;i<N;i++)
            {
                if (ind[i])
                    continue;
                else
                    ind1[i]=0;
            }
        }
        s.insert(mid);
        ver[mid]=ind;
    }
    return st;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...