Submission #611977

#	Time	Username	Problem	Language	Result	Execution time	Memory
611977		M_W	Mechanical Doll (IOI18_doll)	C++17	53 / 100	123 ms	20832 KiB

doll

#include <bits/stdc++.h>
#include "doll.h"
using namespace std;
int N, cnter = 0;
vector<int> adj[100001];
vector<int> X(400002), Y(400002);
vector<int> C(200002);

void build(int a, int cur, int pa, int val, int bit){
    if((1 << (bit + 1)) > adj[a].size() - 1){
      if(adj[a].size() - 1 >= (val | (1 << bit))){
        X[abs(cur)] = adj[a][val] == -1000000 ? pa : adj[a][val];
        Y[abs(cur)] = adj[a][val | (1 << bit)] == -1000000 ? pa : adj[a][val | (1 << bit)];
      }
      else{
        X[abs(cur)] = cur;
        Y[abs(cur)] = adj[a][val] == -1000000 ? pa : adj[a][val];
      }
      return;
    }
    
    if(adj[a].size() - 1 >= (val | (1 << bit))){
      X[abs(cur)] = --cnter;
      build(a, cnter, pa, val, bit + 1);
      Y[abs(cur)] = --cnter;
      build(a, cnter, pa, val | (1 << bit), bit + 1);
    }
    else{
      X[abs(cur)] = cur;
      Y[abs(cur)] = --cnter;
      build(a, cnter, pa, val, bit + 1);
    }
}

void create_circuit(int M, vector<int> A) {
    N = A.size();
    for(int i = 0; i < N - 1; i++) adj[A[i]].push_back(A[i + 1]);
    adj[A[N - 1]].push_back(0);

    C[0] = A[0];
    for(int i = 1; i <= M; i++){
        if(adj[i].size() == 0) continue;
        if(adj[i].size() == 1) C[i] = adj[i][0];
        else{
            C[i] = --cnter;
            int j = 0;
            for(; j <= 20; j++){
              if((1 << j) >= adj[i].size()) break;
            }
            vector<int> tmp;
            for(int k = 0; k < (1 << j) - adj[i].size(); k++) tmp.push_back(-1000000);
            for(auto x : adj[i]) tmp.push_back(x);

            adj[i] = tmp;
            build(i, cnter, cnter, 0, 0);
        }
    }
    vector<int> CA, XA, YA;
    for(int i = 1; i <= abs(cnter); i++){
      XA.push_back(X[i]); YA.push_back(Y[i]);
    }
    for(int i = 0; i <= M; i++) CA.push_back(C[i]);
    answer(CA, XA, YA);
}

Compilation message (stderr)

doll.cpp: In function 'void build(int, int, int, int, int)':
doll.cpp:10:25: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   10 |     if((1 << (bit + 1)) > adj[a].size() - 1){
      |        ~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~
doll.cpp:11:28: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   11 |       if(adj[a].size() - 1 >= (val | (1 << bit))){
      |          ~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~
doll.cpp:22:26: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   22 |     if(adj[a].size() - 1 >= (val | (1 << bit))){
      |        ~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~
doll.cpp: In function 'void create_circuit(int, std::vector<int>)':
doll.cpp:48:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   48 |               if((1 << j) >= adj[i].size()) break;
      |                  ~~~~~~~~~^~~~~~~~~~~~~~~~
doll.cpp:51:30: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   51 |             for(int k = 0; k < (1 << j) - adj[i].size(); k++) tmp.push_back(-1000000);
      |                            ~~^~~~~~~~~~~~~~~~~~~~~~~~~~

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