#include "combo.h"
#include <random>
using namespace std;
// 0 : A, 1 : B, 2 : X, 3 : Y
mt19937 rng(42);
char toBoutons[4] = {'A', 'B', 'X', 'Y'};
bool contient(string S, char interdit)
{
for (int i = 0; i < (int)S.length(); i++)
if (S[i] == interdit)
return true;
return false;
}
string genRandom(int N, string old, char interdit)
{
string S = "";
do
{
S = "";
for (int i = 0; i < N; i++)
S += toBoutons[uniform_int_distribution<int>(0, 3)(rng)];
} while (old[(int)old.length() - N + 1] == S[0] && (!contient(S, interdit) || interdit == '?'));
return S;
}
string extractGood(int N, string S)
{
string res;
for (int i = 0; i < N; i++)
res += S[i];
return res;
}
string guess_sequence(int N)
{
int coinsPrec = 0;
string p = "";
char interdit = '?';
while (coinsPrec < N)
{
p = extractGood(coinsPrec, p) + genRandom(N - coinsPrec, p, interdit);
if (coinsPrec > 0)
interdit = p[0];
coinsPrec = press(p);
}
return p;
}
