Submission #551815

#	Time	Username	Problem	Language	Result	Execution time	Memory
551815		hoanghq2004	Carnival (CEOI14_carnival)	C++14	100 / 100	19 ms	208 KiB

carnival

#include <bits/stdc++.h>
#pragma GCC optimize ("O3")
#pragma GCC optimize ("unroll-loops")
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>

using namespace __gnu_pbds;
using namespace std;

template <typename T>
using ordered_set = tree <T, null_type, less <T>, rb_tree_tag, tree_order_statistics_node_update>;

const int N = 160;

int n, s[N], c[N];

int main() {
    ios :: sync_with_stdio(0); cin.tie(0);
    auto ask = [&](int x, int y) {
        cout << y + 1 << endl;
        for (int i = 1; i <= y; ++i) cout << i << ' ';
        cout << x << endl;
        int z;
        cin >> z;
        return z;
    };
    cin >> n;
    c[1] = ++s[1];
    for (int i = 2; i <= n; ++i) {
        int L = 1, R = i;
        while (R - L > 1) {
            int mid = L + R >> 1;
            if (s[mid] != ask(i, mid)) L = mid;
            else R = mid;
        }
        int j = (s[L] == ask(i, L) ? L : R);
        s[i] = s[i - 1];
        if (j == i) c[i] = ++s[i];
        else c[i] = c[j];
//        cout << j << endl;
    }
    cout << 0 << ' ';
    for (int i = 1; i <= n; ++i) cout << c[i] << ' ';
}

Compilation message (stderr)

carnival.cpp: In function 'int main()':
carnival.cpp:32:25: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   32 |             int mid = L + R >> 1;
      |                       ~~^~~

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