diff --git a/solver.cpp b/solver.cpp index 32013ed..2ba4bc5 100644 --- a/solver.cpp +++ b/solver.cpp @@ -18,6 +18,18 @@ Solver::Solver(int _N, int _M) : N(_N), M(_M) { } vector Solver::guess() { + if(possible.size() == pow(M, N)) { + cout << "Picking first\n"; + vector pick(0); + + int times = N / M + 1; + for(int i = 0; i < times; i++) + for(int j = 0; j < M && pick.size() < N; j++) + pick.push_back(j); + + return pick; + } + return minimax({}).guess; } void Solver::learn(vector guess, Response response) { diff --git a/solver.hpp b/solver.hpp index 951ef54..904b36c 100644 --- a/solver.hpp +++ b/solver.hpp @@ -1,5 +1,6 @@ #pragma once #include +#include #include "global.hpp" using std::set;