# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
912271 | tanprodium | Stove (JOI18_stove) | C++14 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <string>
#include "combo.h"
using namespace std;
namespace {
constexpr int MAX_N = 2000;
constexpr int MAX_NUM_MOVES = 8000;
int N;
std::string S;
int num_moves;
void wrong_answer(const char *MSG) {
printf("Wrong Answer: %s\n", MSG);
exit(0);
}
} // namespace
int press(std::string p) {
if (++num_moves > MAX_NUM_MOVES) {
wrong_answer("too many moves");
}
int len = p.length();
if (len > 4 * N) {
wrong_answer("invalid press");
}
for (int i = 0; i < len; ++i) {
if (p[i] != 'A' && p[i] != 'B' && p[i] != 'X' && p[i] != 'Y') {
wrong_answer("invalid press");
}
}
int coins = 0;
for (int i = 0, j = 0; i < len; ++i) {
if (j < N && S[j] == p[i]) {
++j;
} else if (S[0] == p[i]) {
j = 1;
} else {
j = 0;
}
coins = std::max(coins, j);
}
return coins;
}
string guess_sequence(int N) {
string c = "ABXY";
if (press("AB") >= 1) {
if (press("A") == 1) {
} else {
swap(c[0], c[1]);
}
} else {
if (press("X") == 1) {
swap(c[0], c[2]);
} else {
swap(c[0], c[3]);
}
}
string S = "";
S += c[0];
if (N == 1) return(S);
for (int i = 2; i < N; i++) {
string str = S + c[1] + c[1] + S + c[1] + c[2] + S + c[1] + c[3] + S + c[2];
int sz = (int)S.size();
int taken = press(str);
//cerr << str << ' ' << taken << '\n';
if (taken - sz == 2)
S += c[1];
else if (taken - sz == 1)
S += c[2];
else S += c[3];
}
for (int i = 1; i <= 2; i++) {
string str = S + c[i];
if (press(str) == N) {
S += c[i];
return (S);
}
}
S += c[3];
return (S);
}
int main() {
char buffer[MAX_N + 1];
if (scanf("%s", buffer) != 1) {
fprintf(stderr, "Error while reading input\n");
exit(1);
}
S = buffer;
N = S.length();
num_moves = 0;
std::string answer = guess_sequence(N);
if (answer != S) {
wrong_answer("wrong guess");
exit(0);
}
printf("Accepted: %d\n", num_moves);
return 0;
}