Submission #415108

#	Time	Username	Problem	Language	Result	Execution time	Memory
415108		timmyfeng	Mechanical Doll (IOI18_doll)	C++17	100 / 100	123 ms	9932 KiB

doll

#include <bits/stdc++.h>
using namespace std;

#include "doll.h"

const int N = 400000;

int c[N], x[N], y[N], ext, s;

int build(int k) {
    if (k <= ext) {
        ext -= k;
        return 1;
    } else if (k == 1) {
        return 0;
    }

    int u = ++s;
    x[u - 1] = -build(k / 2);
    y[u - 1] = -build(k / 2);
    return u;
}

void create_circuit(int m, vector<int> a) {
    int n = a.size();

    int k = 1;
    while (k < n) {
        k *= 2;
    }

    ext = k - n;
    int u = build(k);
    fill(c + 1, c + m + 1, -u);

    c[0] = a[0];
    a.push_back(0);
    a.erase(a.begin());

    for (int i = 0; i < n; ++i) {
        int v = u;
        while (x[v - 1] != 0) {
            swap(x[v - 1], y[v - 1]);
            v = -y[v - 1];
        }
        swap(x[v - 1], y[v - 1]);
        y[v - 1] = a[i];
    }

    answer(vector<int>(c, c + m + 1), vector<int>(x, x + s), vector<int>(y, y + s));
}

• Subtask 1: 2 / 2
• Subtask 2: 4 / 4
• Subtask 3: 10 / 10
• Subtask 4: 10 / 10
• Subtask 5: 18 / 18
• Subtask 6: 56 / 56