Submission #444980

#	Time	Username	Problem	Language	Result	Execution time	Memory
444980		BT21tata	사육제 (CEOI14_carnival)	C++17	100 / 100	17 ms	356 KiB

carnival

#include<bits/stdc++.h>
// #pragma GCC target ("avx,avx2,fma")
// #pragma GCC optimize("Ofast")
// #pragma GCC optimize("unroll-loops")
typedef long long ll;
typedef long double ld;
#define SPEED ios_base::sync_with_stdio(false); cin.tie(0), cout.tie(0)
#define rall(v) (v).rbegin(),(v).rend()
#define all(v) (v).begin(),(v).end()
#define OK cerr<<"OK"<<endl<<flush
#define pii pair<int, int>
#define pll pair<ll, ll>
#define pb push_back
#define F first
#define S second
#define y0 jahdakdh
#define y1 jahsadakdakdh
#define endl '\n'
using namespace std;
const ll MOD=1e9+7;
// mt19937 rng(std::chrono::system_clock::now().time_since_epoch().count());

int n, a[155], mx=1, d[155][155], q;
set<int>s;

void ask(int l, int r, int k)
{
    //cout<<l<<' '<<r<<' '<<k<<endl;
    cout<<r-l+2;
    for(int i=l; i<=r; i++)
        cout<<' '<<i;
    cout<<' '<<k<<endl<<flush;
    cin>>q;
}

void cal(int l, int r)
{
    if(d[l][r]) return;
    s.clear();
    for(int i=l; i<=r; i++)
        s.insert(a[i]);
    d[l][r]=s.size();
}

int main()
{
    SPEED;
    cin>>n;
    a[1]=1; d[1][1]=1;
    for(int i=2; i<=n; i++)
    {
        int mid, l=1, r=i-1;
        ask(l, r, i);
        cal(l, r);
        if(q==d[l][r]+1)
        {
            a[i]=++mx;
            continue;
        }
        while(l<r)
        {
            mid=(l+r)>>1;
            ask(l, mid, i);
            cal(l, mid);
            if(q==d[l][mid]+1)
                l=mid+1;
            else r=mid;
        }
        a[i]=a[l];
    }
    for(int i=0; i<=n; i++)
        cout<<a[i]<<' ';
    return 0;
}

• Subtask 1: 20 / 20
• Subtask 2: 20 / 20
• Subtask 3: 20 / 20
• Subtask 4: 20 / 20
• Subtask 5: 20 / 20