Submission #286365

#	Time	Username	Problem	Language	Result	Execution time	Memory
286365		ne4eHbKa	Mechanical Doll (IOI18_doll)	C++17	100 / 100	167 ms	7908 KiB

doll

#include "doll.h"
#include <bits/stdc++.h>
using namespace std;
typedef vector<int> vi;

namespace solution {

vi c, x, y;
int s;
vector<bool> q;

int swicch(int l, int r) {
    int p = l + r >> 1;
    int i = s++;
    x.push_back(0);
    y.push_back(0);
    q.push_back(false);
    if(l + r == 2) {
        if(l) x[i] = -1;
        return -(i + 1);
    }
    if(l >= p) {
        x[i] = -1;
        y[i] = swicch(l - p, r);
        return -(i + 1);
    }
    x[i] = swicch(l, r - p);
    y[i] = swicch(0, p);
    return -(i + 1);
}

void push(int i, int v) {
    int &j = (q[i] ? y[i] : x[i]);
    q[i] = !q[i];
    if(!j) {j = v; return;}
    push(-(j + 1), v);
}

}

void create_circuit (int m, vi a) {
    using namespace solution;
    int n = a.size();
    s = 0;
    x.clear(); x.reserve(n * 4);
    y.clear(); y.reserve(n * 4);
    q.clear(); q.reserve(n * 4);
    if(n == 1) {
        c.assign(m + 1, 0);
        c[0] = a[0];
        answer(c, x, y);
        return;
    }
    c.assign(m + 1, -1);
    c[0] = a[0];
    int e = 0;
    for(; __builtin_popcount(e + n) > 1; ++e);
    swicch(e, n);
    for(int i = 1; i < n; ++i) push(0, a[i]);
    answer(c, x, y);
}

Compilation message (stderr)

doll.cpp: In function 'int solution::swicch(int, int)':
doll.cpp:13:15: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   13 |     int p = l + r >> 1;
      |             ~~^~~

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