游戏
运行代码后，点击"Enable AI Opponent"按钮，AI将作为白棋自动下棋。
玩家作为黑棋，点击棋盘落子。
AI会根据当前棋盘局势自动选择落子位置。
界面
代码
import tkinter as tk
from tkinter import messageboxBOARD_SIZE = 15
CELL_SIZE = 40
STONE_RADIUS = 15
BOARD_COLOR = "#DEB887"class Gomoku:def __init__(self, root):self.root = rootself.root.title("Gomoku")self.root.resizable(False, False)self.board = [[0] * BOARD_SIZE for _ in range(BOARD_SIZE)] # 0: empty, 1: black, 2: whiteself.is_black_turn = True # Black goes firstself.ai_enabled = False # Whether AI opponent is enabledself.canvas = tk.Canvas(root, width=BOARD_SIZE * CELL_SIZE, height=BOARD_SIZE * CELL_SIZE, bg=BOARD_COLOR)self.canvas.pack()self.draw_board()self.canvas.bind("<Button-1>", self.on_mouse_click)# Add AI toggle buttonself.ai_button = tk.Button(root, text="Enable AI Opponent", command=self.toggle_ai)self.ai_button.pack()def draw_board(self):self.canvas.delete("all")for i in range(BOARD_SIZE):self.canvas.create_line(0, i * CELL_SIZE, BOARD_SIZE * CELL_SIZE, i * CELL_SIZE) # Draw horizontal linesself.canvas.create_line(i * CELL_SIZE, 0, i * CELL_SIZE, BOARD_SIZE * CELL_SIZE) # Draw vertical linesfor i in range(BOARD_SIZE):for j in range(BOARD_SIZE):if self.board[i][j] == 1:self.canvas.create_oval(i * CELL_SIZE - STONE_RADIUS, j * CELL_SIZE - STONE_RADIUS,i * CELL_SIZE + STONE_RADIUS, j * CELL_SIZE + STONE_RADIUS, fill="black")elif self.board[i][j] == 2:self.canvas.create_oval(i * CELL_SIZE - STONE_RADIUS, j * CELL_SIZE - STONE_RADIUS,i * CELL_SIZE + STONE_RADIUS, j * CELL_SIZE + STONE_RADIUS, fill="white")def on_mouse_click(self, event):if self.is_black_turn or not self.ai_enabled:x = (event.x + CELL_SIZE // 2) // CELL_SIZEy = (event.y + CELL_SIZE // 2) // CELL_SIZEif 0 <= x < BOARD_SIZE and 0 <= y < BOARD_SIZE and self.board[x][y] == 0:self.place_stone(x, y)if self.check_win(x, y):winner = "Black" if self.is_black_turn else "White"messagebox.showinfo("Game Over", f"{winner} wins!")self.reset_board()else:self.is_black_turn = not self.is_black_turnif self.ai_enabled and not self.is_black_turn:self.ai_move()def place_stone(self, x, y):self.board[x][y] = 1 if self.is_black_turn else 2self.draw_board()def reset_board(self):self.board = [[0] * BOARD_SIZE for _ in range(BOARD_SIZE)]self.is_black_turn = Trueself.draw_board()def check_win(self, x, y):player = self.board[x][y]return self.check_direction(x, y, 1, 0, player) or \self.check_direction(x, y, 0, 1, player) or \self.check_direction(x, y, 1, 1, player) or \self.check_direction(x, y, 1, -1, player)def check_direction(self, x, y, dx, dy, player):count = 1for i in range(1, 5):nx, ny = x + i * dx, y + i * dyif 0 <= nx < BOARD_SIZE and 0 <= ny < BOARD_SIZE and self.board[nx][ny] == player:count += 1else:breakfor i in range(1, 5):nx, ny = x - i * dx, y - i * dyif 0 <= nx < BOARD_SIZE and 0 <= ny < BOARD_SIZE and self.board[nx][ny] == player:count += 1else:breakreturn count >= 5def toggle_ai(self):self.ai_enabled = not self.ai_enabledself.ai_button.config(text="Disable AI Opponent" if self.ai_enabled else "Enable AI Opponent")if self.ai_enabled and not self.is_black_turn:self.ai_move()def ai_move(self):best_score = -1best_move = Nonefor i in range(BOARD_SIZE):for j in range(BOARD_SIZE):if self.board[i][j] == 0:score = self.evaluate_position(i, j)if score > best_score:best_score = scorebest_move = (i, j)if best_move:self.place_stone(*best_move)if self.check_win(*best_move):winner = "White"messagebox.showinfo("Game Over", f"{winner} wins!")self.reset_board()else:self.is_black_turn = not self.is_black_turndef evaluate_position(self, x, y):# Simple scoring systemscore = 0for dx, dy in [(1, 0), (0, 1), (1, 1), (1, -1)]:score += self.evaluate_direction(x, y, dx, dy, 2) # AI is Whitescore += self.evaluate_direction(x, y, dx, dy, 1) # Block Blackreturn scoredef evaluate_direction(self, x, y, dx, dy, player):count = 0for i in range(1, 5):nx, ny = x + i * dx, y + i * dyif 0 <= nx < BOARD_SIZE and 0 <= ny < BOARD_SIZE and self.board[nx][ny] == player:count += 1else:breakfor i in range(1, 5):nx, ny = x - i * dx, y - i * dyif 0 <= nx < BOARD_SIZE and 0 <= ny < BOARD_SIZE and self.board[nx][ny] == player:count += 1else:breakreturn count ** 2 # More consecutive stones, higher scoreif __name__ == "__main__":root = tk.Tk()game = Gomoku(root)root.mainloop()