oj.uz

Submission #1074281

# Submission time Handle Problem Language Result Execution time Memory
1074281 2024-08-25T09:21:00 Z Elias Mechanical Doll (IOI18_doll) C++17
100 / 100
 110 ms 19760 KB 
#include <bits/stdc++.h>

using namespace std;

#ifndef _DEBUG
#include "doll.h"
#else
void answer(std::vector<int> C, std::vector<int> X, std::vector<int> Y);
#endif

void wrong(const char *MSG)
{
	printf("Wrong Answer: %s\n", MSG);
	assert(false);
	exit(0);
}

void sim(vector<int> A, vector<int> c, vector<int> x, vector<int> y, int N, int M)
{
	int S = x.size();
	auto IC = c;
	auto IX = x;
	auto IY = y;
	constexpr int p_MAX = 20000000;
	constexpr int s_MAX = 400000;

	if (S > s_MAX)
	{
		char str[50];
		sprintf(str, "over %d switches", s_MAX);
		wrong(str);
	}
	for (int i = 0; i <= M; ++i)
	{
		if (!(-S <= IC[i] && IC[i] <= M))
		{
			wrong("wrong serial number");
		}
	}
	for (int j = 1; j <= S; ++j)
	{
		if (!(-S <= IX[j - 1] && IX[j - 1] <= M))
		{
			wrong("wrong serial number");
		}
		if (!(-S <= IY[j - 1] && IY[j - 1] <= M))
		{
			wrong("wrong serial number");
		}
	}

	int P = 0;
	std::vector<bool> state(S + 1, false);
	int pos = IC[0];
	int k = 0;
	FILE *file_log = fopen("log.txt", "w");
	fprintf(file_log, "0\n");
	for (;;)
	{
		fprintf(file_log, "%d\n", pos);
		if (pos < 0)
		{
			if (++P > p_MAX)
			{
				fclose(file_log);
				char str[50];
				sprintf(str, "over %d inversions", p_MAX);
				wrong(str);
			}
			state[-pos] = !state[-pos];
			pos = state[-pos] ? IX[-(1 + pos)] : IY[-(1 + pos)];
		}
		else
		{
			if (pos == 0)
			{
				break;
			}
			if (k >= N)
			{
				fclose(file_log);
				wrong("wrong motion");
			}
			if (pos != A[k++])
			{
				fclose(file_log);
				wrong("wrong motion");
			}
			pos = IC[pos];
		}
	}
	fclose(file_log);
	if (k != N)
	{
		wrong("wrong motion");
	}
	for (int j = 1; j <= S; ++j)
	{
		if (state[j])
		{
			wrong("state 'Y'");
		}
	}
	// printf("Accepted: %d %d\n", S, P);
}


int n, m;
vector<int> a;
vector<vector<int>> pos;
int cur = 1;

vector<int> c, x, y;

int kgl, pgl;

void build(int i, int l, int r)
{
	int mid = (l + r) / 2;
	if (mid <= pgl)
		x[-i - 1] = -1;
	else
	{
		if (l != mid)
		{
			x[-i - 1] = -cur;
			cur++;
			x.push_back(0);
			y.push_back(0);
			build(x[-i - 1], l, mid);
		}
		else
		{
			x[-i - 1] = 3 * n;
		}
	}
	if (mid + 1 != r)
	{
		y[-i - 1] = -cur;
		cur++;
		x.push_back(0);
		y.push_back(0);
		build(y[-i - 1], mid + 1, r);
	}
	else
	{
		y[-i - 1] = 3 * n;
	}
}

void build_sgt(int nod)
{
	cur = 2;
	kgl = __lg(n);
	if ((1 << kgl) < n)
		kgl++;
	pgl = (1 << kgl) - (int)n;
	build(nod, 1, (1 << kgl));
}

int cr = 0;
vector<int> vis;

void fix(int i)
{
	if (i == 0)
		return;
	if (i >= 1)
	{
		cr++;
		fix(c[i]);
		return;
	}

	if (vis[-i - 1] == 0)
	{
		vis[-i - 1] = 1;
		if (x[-i - 1] > 2 * n)
		{
			x[-i - 1] = a[cr];
		}
		fix(x[-i - 1]);
	}
	else
	{
		vis[-i - 1] = 0;
		if (y[-i - 1] > 2 * n)
		{
			y[-i - 1] = a[cr];
		}
		fix(y[-i - 1]);
	}
}

void create_circuit(int M, vector<int> A)
{
	A.push_back(0);
	n = A.size();
	a = A;
	m = M;
	pos.resize(m + 1);
	
	for (int i = 0; i < a.size(); i++)
		pos[a[i]].push_back(i);

	a.push_back(0);
	c.resize(m + 1);
	for (int i = 0; i <= m; i++)
		c[i] = -1;

	x.push_back(0);
	y.push_back(0);
	vis.resize(2 * n);
	build_sgt(-1);

	fix(-1);

	answer(c, x, y);
}

#ifdef _DEBUG

namespace
{

	constexpr int P_MAX = 20000000;
	constexpr int S_MAX = 400000;

	int M, N;
	std::vector<int> A;

	bool answered;
	int S;
	std::vector<int> IC, IX, IY;

	int read_int()
	{
		int x;
		if (scanf("%d", &x) != 1)
		{
			fprintf(stderr, "Error while reading input\n");
			exit(1);
		}
		return x;
	}

	void wrong_answer(const char *MSG)
	{
		printf("Wrong Answer: %s\n", MSG);
		exit(0);
	}

	void simulate()
	{
		if (S > S_MAX)
		{
			char str[50];
			sprintf(str, "over %d switches", S_MAX);
			wrong_answer(str);
		}
		for (int i = 0; i <= M; ++i)
		{
			if (!(-S <= IC[i] && IC[i] <= M))
			{
				wrong_answer("wrong serial number");
			}
		}
		for (int j = 1; j <= S; ++j)
		{
			if (!(-S <= IX[j - 1] && IX[j - 1] <= M))
			{
				wrong_answer("wrong serial number");
			}
			if (!(-S <= IY[j - 1] && IY[j - 1] <= M))
			{
				wrong_answer("wrong serial number");
			}
		}

		int P = 0;
		std::vector<bool> state(S + 1, false);
		int pos = IC[0];
		int k = 0;
		FILE *file_log = fopen("log.txt", "w");
		fprintf(file_log, "0\n");
		for (;;)
		{
			fprintf(file_log, "%d\n", pos);
			if (pos < 0)
			{
				if (++P > P_MAX)
				{
					fclose(file_log);
					char str[50];
					sprintf(str, "over %d inversions", P_MAX);
					wrong_answer(str);
				}
				state[-pos] = !state[-pos];
				pos = state[-pos] ? IX[-(1 + pos)] : IY[-(1 + pos)];
			}
			else
			{
				if (pos == 0)
				{
					break;
				}
				if (k >= N)
				{
					fclose(file_log);
					wrong_answer("wrong motion");
				}
				if (pos != A[k++])
				{
					fclose(file_log);
					wrong_answer("wrong motion");
				}
				pos = IC[pos];
			}
		}
		fclose(file_log);
		if (k != N)
		{
			wrong_answer("wrong motion");
		}
		for (int j = 1; j <= S; ++j)
		{
			if (state[j])
			{
				wrong_answer("state 'Y'");
			}
		}
		printf("Accepted: %d %d\n", S, P);
	}

} // namespace

void answer(std::vector<int> C, std::vector<int> X, std::vector<int> Y)
{
	if (answered)
	{
		wrong_answer("answered not exactly once");
	}
	answered = true;
	// check if input format is correct
	if ((int)C.size() != M + 1)
	{
		wrong_answer("wrong array length");
	}
	if (X.size() != Y.size())
	{
		wrong_answer("wrong array length");
	}
	S = X.size();
	IC = C;
	IX = X;
	IY = Y;
}

int main()
{
	M = read_int();
	N = read_int();
	A.resize(N);
	for (int k = 0; k < N; ++k)
	{
		A[k] = read_int();
	}

	answered = false;
	create_circuit(M, A);
	if (!answered)
	{
		wrong_answer("answered not exactly once");
	}
	FILE *file_out = fopen("out.txt", "w");
	fprintf(file_out, "%d\n", S);
	for (int i = 0; i <= M; ++i)
	{
		fprintf(file_out, "%d\n", IC[i]);
	}
	for (int j = 1; j <= S; ++j)
	{
		fprintf(file_out, "%d %d\n", IX[j - 1], IY[j - 1]);
	}
	fclose(file_out);
	simulate();
	return 0;
}

#endif

Compilation message

doll.cpp: In function 'void create_circuit(int, std::vector<int>)':
doll.cpp:203:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  203 |  for (int i = 0; i < a.size(); i++)
      |                  ~~^~~~~~~~~~

Subtask #1
2.0 / 2.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 38 ms 10060 KB Output is correct
3 Correct 32 ms 9036 KB Output is correct
4 Correct 1 ms 348 KB Output is correct
5 Correct 6 ms 3932 KB Output is correct
6 Correct 54 ms 13148 KB Output is correct
7 Correct 0 ms 344 KB Output is correct

Subtask #2
4.0 / 4.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 38 ms 10060 KB Output is correct
3 Correct 32 ms 9036 KB Output is correct
4 Correct 1 ms 348 KB Output is correct
5 Correct 6 ms 3932 KB Output is correct
6 Correct 54 ms 13148 KB Output is correct
7 Correct 0 ms 344 KB Output is correct
8 Correct 67 ms 13332 KB Output is correct
9 Correct 71 ms 14812 KB Output is correct
10 Correct 92 ms 19760 KB Output is correct
11 Correct 1 ms 344 KB Output is correct
12 Correct 0 ms 344 KB Output is correct
13 Correct 0 ms 348 KB Output is correct

Subtask #3
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 38 ms 10060 KB Output is correct
3 Correct 32 ms 9036 KB Output is correct
4 Correct 1 ms 348 KB Output is correct
5 Correct 6 ms 3932 KB Output is correct
6 Correct 54 ms 13148 KB Output is correct
7 Correct 0 ms 344 KB Output is correct
8 Correct 67 ms 13332 KB Output is correct
9 Correct 71 ms 14812 KB Output is correct
10 Correct 92 ms 19760 KB Output is correct
11 Correct 1 ms 344 KB Output is correct
12 Correct 0 ms 344 KB Output is correct
13 Correct 0 ms 348 KB Output is correct
14 Correct 95 ms 17200 KB Output is correct
15 Correct 63 ms 11084 KB Output is correct
16 Correct 93 ms 16012 KB Output is correct
17 Correct 0 ms 344 KB Output is correct
18 Correct 0 ms 348 KB Output is correct
19 Correct 0 ms 348 KB Output is correct
20 Correct 110 ms 17864 KB Output is correct
21 Correct 0 ms 348 KB Output is correct
22 Correct 0 ms 348 KB Output is correct

Subtask #4
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 1 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 1 ms 344 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 1 ms 348 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Correct 0 ms 348 KB Output is correct

Subtask #5
18.0 / 18.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 52 ms 9068 KB Output is correct
3 Correct 55 ms 8692 KB Output is correct
4 Correct 75 ms 13364 KB Output is correct

Subtask #6
56.0 / 56.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 52 ms 9068 KB Output is correct
3 Correct 55 ms 8692 KB Output is correct
4 Correct 75 ms 13364 KB Output is correct
5 Correct 97 ms 16432 KB Output is correct
6 Correct 97 ms 15548 KB Output is correct
7 Correct 96 ms 15924 KB Output is correct
8 Correct 97 ms 15156 KB Output is correct
9 Correct 56 ms 8736 KB Output is correct
10 Correct 86 ms 14900 KB Output is correct
11 Correct 90 ms 14600 KB Output is correct
12 Correct 57 ms 9420 KB Output is correct
13 Correct 65 ms 10328 KB Output is correct
14 Correct 69 ms 10436 KB Output is correct
15 Correct 92 ms 10696 KB Output is correct
16 Correct 2 ms 600 KB Output is correct
17 Correct 47 ms 9600 KB Output is correct
18 Correct 55 ms 9256 KB Output is correct
19 Correct 55 ms 9424 KB Output is correct
20 Correct 81 ms 14132 KB Output is correct
21 Correct 86 ms 14232 KB Output is correct
22 Correct 75 ms 13872 KB Output is correct