#include "doll.h"
#ifdef LOCAL
//#ifdef AMOGUS
#include <cstdio>
#include <cstdlib>
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(1);
}
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 = int(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
#include <ios>
#include <vector>
#include <cassert>
#define REP(i, n) for(int i=0; i<(n); ++i)
#define FOR(i, p, n) for(int i=(p); i<=(n); ++i)
#define RFOR(i, p, n) for(int i=(p); i>=(n); --i)
#define V std::vector
#define F std::function
#define sz(A) int(A.size())
#define all(A) A.begin(), A.end()
typedef long long ll;
typedef V <int> vi;
typedef V <ll> vll;
typedef V <bool> vb;
void create_circuit(int M, vi A) {
int on=sz(A);
while (sz(A)&(sz(A)+1))
A.emplace_back(-1);
A.emplace_back(0);
int n = sz(A),comp;
for (comp=1; comp<n; comp<<=1);
assert(comp==n);
int s=comp-1;
vi X(s+1, -1),Y(s+1, -1);
comp>>=1,--comp;
FOR(i, 1, comp)
X[i]=i<<1,Y[i]=(i<<1)+1;
vi ile(s+1, 0);
for (int i : A){
int a=1;
while (a<=comp){
int b=(a<<1)+(ile[a]&1);
++ile[a];
a=b;
}
assert(ile[a]<2);
if (ile[a]&1)
Y[a]=i;
else
X[a]=i;
//printf("a=%d a-comp=%d\n", a, a-comp);
ile[a]^=1;
}
FOR(i, 1, s)
assert(!(ile[i]&1));
vi C(M+1, -1); // jeszcze 0
FOR(i, 1, comp)
X[i]*=-1,Y[i]*=-1;
X.erase(X.begin()),Y.erase(Y.begin());
//assert(sz(X)<=2*on);
answer(C, X, Y);
}
Compilation message
doll.cpp: In function 'void create_circuit(int, vi)':
doll.cpp:160:9: warning: unused variable 'on' [-Wunused-variable]
160 | int on=sz(A);
| ^~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
0 ms |
424 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
0 ms |
424 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
0 ms |
424 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
348 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Partially correct |
1 ms |
348 KB |
Output is partially correct |
2 |
Partially correct |
68 ms |
9824 KB |
Output is partially correct |
3 |
Partially correct |
64 ms |
9696 KB |
Output is partially correct |
4 |
Partially correct |
77 ms |
10616 KB |
Output is partially correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Partially correct |
1 ms |
348 KB |
Output is partially correct |
2 |
Partially correct |
68 ms |
9824 KB |
Output is partially correct |
3 |
Partially correct |
64 ms |
9696 KB |
Output is partially correct |
4 |
Partially correct |
77 ms |
10616 KB |
Output is partially correct |
5 |
Partially correct |
81 ms |
11076 KB |
Output is partially correct |
6 |
Partially correct |
73 ms |
10820 KB |
Output is partially correct |
7 |
Partially correct |
69 ms |
10816 KB |
Output is partially correct |
8 |
Partially correct |
76 ms |
10520 KB |
Output is partially correct |
9 |
Partially correct |
60 ms |
9824 KB |
Output is partially correct |
10 |
Partially correct |
69 ms |
10564 KB |
Output is partially correct |
11 |
Partially correct |
68 ms |
10564 KB |
Output is partially correct |
12 |
Partially correct |
62 ms |
9808 KB |
Output is partially correct |
13 |
Partially correct |
81 ms |
9816 KB |
Output is partially correct |
14 |
Partially correct |
68 ms |
9812 KB |
Output is partially correct |
15 |
Partially correct |
67 ms |
10064 KB |
Output is partially correct |
16 |
Partially correct |
2 ms |
604 KB |
Output is partially correct |
17 |
Correct |
34 ms |
6536 KB |
Output is correct |
18 |
Partially correct |
62 ms |
9812 KB |
Output is partially correct |
19 |
Partially correct |
63 ms |
9624 KB |
Output is partially correct |
20 |
Partially correct |
67 ms |
10564 KB |
Output is partially correct |
21 |
Partially correct |
68 ms |
10636 KB |
Output is partially correct |
22 |
Partially correct |
75 ms |
10680 KB |
Output is partially correct |