人机对弈五子棋python实现代码

以下是一个简单的人机对弈五子棋Python实现代码：

import numpy as np
import pygame

# 初始化Pygame
pygame.init()

# 设置棋盘大小和棋子大小
board_size = 600
piece_size = 30

# 设置行列数
rows = board_size // piece_size
cols = board_size // piece_size

# 设置颜色
black = (0, 0, 0)
white = (255, 255, 255)
gray = (192, 192, 192)
blue = (0, 0, 255)

# 创建棋盘
board = np.zeros((rows, cols))

# 设置字体
font = pygame.font.Font(None, 30)

# 创建屏幕
screen = pygame.display.set_mode((board_size, board_size))
pygame.display.set_caption('人机对弈五子棋')


def draw_board():
    # 绘制棋盘背景
    screen.fill(gray)

    # 绘制棋盘网格
    for i in range(rows):
        pygame.draw.line(screen, black, (0, i * piece_size), (board_size, i * piece_size))
        pygame.draw.line(screen, black, (i * piece_size, 0), (i * piece_size, board_size))

    # 绘制棋子
    for i in range(rows):
        for j in range(cols):
            if board[i][j] == 1:
                pygame.draw.circle(screen, black, (i * piece_size + piece_size // 2, j * piece_size + piece_size // 2), piece_size // 2 - 2)
            elif board[i][j] == 2:
                pygame.draw.circle(screen, white, (i * piece_size + piece_size // 2, j * piece_size + piece_size // 2), piece_size // 2 - 2)

    # 更新屏幕
    pygame.display.update()


def get_winner(board):
    # 检查每个方向上是否有五个棋子相连
    for i in range(rows):
        for j in range(cols):
            if j + 4 < cols and board[i][j] == board[i][j + 1] == board[i][j + 2] == board[i][j + 3] == board[i][j + 4]:
                return board[i][j]
            elif i + 4 < rows and board[i][j] == board[i + 1][j] == board[i + 2][j] == board[i + 3][j] == board[i + 4][j]:
                return board[i][j]
            elif i + 4 < rows and j + 4 < cols and board[i][j] == board[i + 1][j + 1] == board[i + 2][j + 2] == board[i + 3][j + 3] == board[i + 4][j + 4]:
                return board[i][j]
            elif i + 4 < rows and j - 4 >= 0 and board[i][j] == board[i + 1][j - 1] == board[i + 2][j - 2] == board[i + 3][j - 3] == board[i + 4][j - 4]:
                return board[i][j]

    # 如果没有五个棋子相连，则返回0
    return 0


def get_next_move(board, player):
    # 随机生成下一步棋子的位置
    while True:
        row = np.random.randint(rows)
        col = np.random.randint(cols)
        if board[row][col] == 0:
            return row, col


# 设置初始玩家为黑色
player = 1

# 游戏循环
while True:
    # 处理事件
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            quit()
        elif event.type == pygame.MOUSEBUTTONDOWN:
            # 获取鼠标点击位置
            x, y = pygame.mouse.get_pos()
            row = x // piece_size
            col = y // piece_size
            if board[row][col] == 0:
                # 更新棋盘状态
                board[row][col] = player

                # 检查是否有玩家获胜
                winner = get_winner(board)
                if winner == player:
                    text = font.render('你赢了！', True, blue)
                    screen.blit(text, (board_size - 120, 10))
                    pygame.display.update()
                    pygame.time.wait(3000)
                    pygame.quit()
                    quit()

                # 更新屏幕
                draw_board()

                # 切换玩家
                player = 2 if player == 1 else 1

                # 让计算机下一步棋子
                row, col = get_next_move(board, player)
                board[row][col] = player

                # 检查是否有计算机获胜
                winner = get_winner(board)
                if winner == player:
                    text = font.render('计算机赢了！', True, blue)
                    screen.blit(text, (board_size - 150, 10))
                    pygame.display.update()
                    pygame.time.wait(3000)
                    pygame.quit()
                    quit()

                # 更新屏幕
                draw_board()

                # 切换玩家
                player = 2 if player == 1 else 1

这个代码实现了一个简单的人机对弈五子棋游戏，玩家可以使用鼠标在棋盘上下棋，计算机会随机生成下一步棋子的位置。如果有一方获胜，则会在屏幕上显示相应的提示。