Я пытался реализовать минимакс для простой игры ti c -ta c -toe, но пока она работает, я не смог заставить ее дать правильный результат. Функция предполагает, что AI играет как 'O', а игрок играет как 'X'.
import itertools
class Board:
def __init__(self):
self.grid = {}
def get_symbol(self, move):
return self.grid.get(move)
def set_symbol(self, move, symbol):
self.grid[move] = symbol
def clear_symbol(self, move):
self.grid.pop(move)
def has_winner(self):
return self.is_winner('X') or self.is_winner('O')
def is_winner(self, symbol):
wins = [[(0, 0), (1, 0), (2, 0)],
[(0, 1), (1, 1), (2, 1)],
[(0, 2), (1, 2), (2, 2)],
[(0, 0), (0, 1), (0, 2)],
[(1, 0), (1, 1), (1, 2)],
[(2, 0), (2, 1), (2, 2)],
[(0, 0), (1, 1), (2, 2)],
[(0, 2), (1, 1), (2, 0)]]
for win in wins:
if all(self.get_symbol(e) == symbol for e in win):
return True
return False
def is_full(self):
return all(self.get_symbol(tuple(e)) is not None for e in itertools.product(range(3), repeat=2))
def get_moves(self):
return [move for move in itertools.product(range(3), repeat=2) if move not in self.grid.keys()]
def get_heuristic(self):
if self.is_winner('X'):
return -1
if self.is_winner('O'):
return 1
return 0
def get_enemy(symbol):
if symbol == 'O':
return 'X'
return 'O'
def is_terminal(board):
return board.has_winner() or board.is_full()
def minimax(board, depth, piece):
if depth == 0 or is_terminal(board):
return board.get_heuristic()
if piece == 'O':
value = float('-inf')
for move in board.get_moves():
board.set_symbol(move, piece)
value = max(value, minimax(board, depth-1, get_enemy(piece)))
board.clear_symbol(move)
return value
else:
value = float('inf')
for move in board.get_moves():
board.set_symbol(move, piece)
value = min(value, minimax(board, depth-1, get_enemy(piece)))
board.clear_symbol(move)
return value
def main():
board = Board()
board.set_symbol((0, 0), 'X')
board.set_symbol((0, 1), 'O')
board.set_symbol((1, 0), 'X')
moves = board.get_moves()
scores = []
for move in moves:
board.set_symbol(move, 'O')
scores.append(minimax(board, 100, 'O'))
board.clear_symbol(move)
m = max(scores)
for i, e in enumerate(moves):
if scores[i] == m:
print(e)
if __name__ == '__main__':
main()
Этот должен print '(2, 0)', так как любая другая позиция заставит 'X' победить в следующем ходу, однако вместо этого она напечатает различные подмножества из 6 возможных результатов в зависимости от того, какая глубина передана минимаксу. Я попытался использовать 'X' и 'O' в качестве начального аргумента для части к минимаксной функции, но ни один не дает правильных результатов Что не так?