Submission #75619

#	Time	Username	Problem	Language	Result	Execution time	Memory
75619		gs14004	Mechanical Doll (IOI18_doll)	C++17	100 / 100	89 ms	10296 KiB

doll

#include "doll.h"
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 400005;

int vtx_number, total_leaf;
int X[MAXN], Y[MAXN], mp[MAXN];

void dfs(int s, int e, int v, int d){
	vtx_number++;
	if(e - s == 1){
		if(s < total_leaf) X[vtx_number - 1] = -1;
		else X[vtx_number - 1] = v;
		Y[vtx_number - 1] = v ^ (1 << d);
		return;
	}
	int cur_vtx = vtx_number - 1;
	int m = (s + e) / 2;
	if(m < total_leaf){
		X[cur_vtx] = -1;
	}
	else{
		X[cur_vtx] = -vtx_number - 1;
		dfs(s, m, v, d + 1);
	}
	Y[cur_vtx] = -vtx_number - 1;
	dfs(m+1, e, v ^ (1 << d), d + 1);
}

void create_circuit(int M, std::vector<int> A) {
	if(A.size() == 1){
		vector<int> C(M + 1, 0), X, Y;
		C[0] = A[0];
		answer(C, X, Y);
		return;
	}
	vector<int> C(M + 1, -1);
	C[0] = A[0];
	int K = 0;
	while((1 << K) < A.size()) K++;
	total_leaf = (1 << K) - A.size();
	A.erase(A.begin());
	A.push_back(0);
	dfs(0, (1 << K) - 1, 0, 0);
	for(int i=0; i<vtx_number; i++){
		if(X[i] >= 0) mp[X[i]] = 1;
		if(Y[i] >= 0) mp[Y[i]] = 1;
	}
	int ptr = 0;
	for(int i=0; i<(1<<K); i++){
		if(mp[i]){
			mp[i] = A[ptr++];
		}
	}
	for(int i=0; i<vtx_number; i++){
		if(X[i] >= 0) X[i] = mp[X[i]];
		if(Y[i] >= 0) Y[i] = mp[Y[i]];
	}
	vector<int> vx(X, X + vtx_number);
	vector<int> vy(Y, Y + vtx_number);
	answer(C, vx, vy);
}

Compilation message (stderr)

doll.cpp: In function 'void create_circuit(int, std::vector<int>)':
doll.cpp:40:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   40 |  while((1 << K) < A.size()) K++;
      |        ~~~~~~~~~^~~~~~~~~~

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