Submission #413424

#	Time	Username	Problem	Language	Result	Execution time	Memory
413424		schse	Mechanical Doll (IOI18_doll)	C++17	16 / 100	126 ms	8940 KiB

doll

#include "doll.h"
#ifndef EVAL
#include "grader.cpp"
#endif
#include <bits/stdc++.h>
using namespace std;

vector<int> C, X(15), Y(15);

vector<bool> tree(16);
void rec(int index, int value)
{
  if (index > 7)
  {
    if (tree[index])
      X[index - 1] = value;
    else
      Y[index - 1] = value;
  }
  else if (tree[index])
    rec(index * 2, value);
  else
    rec(index * 2 + 1, value);
  tree[index] = !tree[index];
}

void create_circuit(int M, std::vector<int> A)
{

  int N = A.size();
  fill(tree.begin(), tree.end(), true);
  if (N == 16)
  {
    C.resize(M + 1);
    for (int i = 1; i < 8; i++)
    {
      X[i - 1] = i * -2;
      Y[i - 1] = i * -2 - 1;
    }
    for (int i = 1; i < N; i++)
      rec(1, A[i]);
    rec(1, 0);
    fill(C.begin(), C.end(), -1);
    C[0] = A[0];
  }
  else
  {
    vector<vector<int>> g(M + 1);
    C.resize(M + 1);
    X = Y = {};
    fill(C.begin(), C.end(), 0);
    C[0] = A[0];
    g[0].push_back(A[0]);

    for (int i = 1; i < N; i++)
      g[A[i - 1]].push_back(A[i]);
    g[A[N - 1]].push_back(0);

    for (int i = 0; i < M + 1; i++)
    {
      if (g[i].size() == 0)
        C[i] = 0;
      else if (g[i].size() == 1)
        C[i] = g[i][0];
      else if (g[i].size() == 2)
      {
        X.push_back(g[i][0]);
        Y.push_back(g[i][1]);
        C[i] = X.size() * -1;
      }
    }
  }
  answer(C, X, Y);
}

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