Submission #1081768

#	Time	Username	Problem	Language	Result	Execution time	Memory
1081768		erray	Mechanical Doll (IOI18_doll)	C++17	100 / 100	107 ms	10196 KiB

doll

#include "doll.h"
#include <bits/stdc++.h>

using namespace std;

#ifdef DEBUG 
  #include "/home/ioi/contests/ioi18_d2/debug.h"
#else 
  #define debug(...) void(37)
#endif



void create_circuit(int M, std::vector<int> A) {
  A.push_back(0);
  int N = A.size();
  /*
  std::vector<int> C(M + 1);
  C[0] = -1;
  for (int i = 1; i <= M; ++i) {
    C[i] = 1;
  }
  std::vector<int> X(N), Y(N);
  for (int k = 0; k < N; ++k) {
    X[k] = Y[k] = A[k];
  }
  answer(C, X, Y);
  */
  int S = 1;
  while (S < N) {
    S *= 2;
  }
  vector<int> X, Y;
  auto Dfs = [&](int l, int r, auto&& Dfs) {
    if (S - r > N) {
      return -1;
    }
    if (l == r) { 
      return 0;
    }
    int mid = (l + r) / 2;
    int v = int(X.size());
    X.push_back(-1);
    Y.push_back(-1);
    int left = Dfs(l, mid, Dfs);
    int right = Dfs(mid + 1, r, Dfs);
    X[v] = left, Y[v] = right;
    return -1 - v;
  };
  Dfs(0, S - 1, Dfs);
  int n = int(X.size());
  vector<int> p(n);
  for (int i = 0; i < N; ++i) {
    int v = 0;
    while (true) {
      int to = (p[v] ? Y[v] : X[v]);
      p[v] ^= 1;
      if (to == 0) {
        break;
      }
      v = -to - 1;
    }
    (p[v] ? X[v] : Y[v]) = A[i];
  }
  answer(vector<int>(M + 1, -1), X, Y);
}

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