基于α-β剪枝python实现五子棋人机对战pygame

import pygame import pygame.ftfont from pygame import * import sys import button import random import re import copy colors = { "white": (255, 255, 255), "black": (0, 0, 0) } r_lines = [ ((20, 20), (20, 620)), ((20, 620), (620, 620)), ((20, 20), (620, 20)), ((620, 20), (620, 620)) ] # 棋形匹配 l = [[] for i in range(5)] l[4] = ["11111"] l[3] = ["011110", "211110"] l[2] = ["11101", "01110", "211100", "2011102"] l[1] = ["011010", "11011", "211010", "210110", "11001", "001100", "211000"] l[0] = ["10101", "01010", "010010", "210100", "210010", "10001"] # 棋形与分值索引 values = { # 连五 "11111": 999900000, # 活四 "011110": 33300000, # 冲四 "211110": 6250000, "11101": 6250000, "11011": 6250000, # 活三 "01110": 625000, "011010": 625000, # 眠三 "211100": 12500, "211010": 12500, "210110": 12500, "11001": 12500, "10101": 12500, "2011102": 12500, # 活二 "001100": 250, "01010": 250, "010010": 250, # 眠二 "211000": 25, "210100": 25, "210010": 25, "10001": 25, # 其他 "others": 8 } # 位置得分， 越靠近中心位置分越高 values_of_pos = [ [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [0, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0], [0, 2, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 2, 0], [0, 2, 4, 6, 6, 6, 6, 6, 6, 6, 6, 6, 4, 2, 0], [0, 2, 4, 6, 7, 7, 7, 7, 7, 7, 7, 6, 4, 2, 0], [0, 2, 4, 6, 7, 8, 8, 8, 8, 8, 7, 6, 4, 2, 0], [0, 2, 4, 6, 7, 8, 9, 9, 9, 8, 7, 6, 4, 2, 0], [0, 2, 4, 6, 7, 8, 9, 10, 9, 8, 7, 6, 4, 2, 0], [0, 2, 4, 6, 7, 8, 9, 9, 9, 8, 7, 6, 4, 2, 0], [0, 2, 4, 6, 7, 8, 8, 8, 8, 8, 7, 6, 4, 2, 0], [0, 2, 4, 6, 7, 7, 7, 7, 7, 7, 7, 6, 4, 2, 0], [0, 2, 4, 6, 6, 6, 6, 6, 6, 6, 6, 6, 4, 2, 0], [0, 2, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 2, 0], [0, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 0], [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], ] me_first = 1 ai_first = 0 # 设置先后手规则 first = me_first # 默认先手为玩家， 即玩家持黑 Myturn = 1 # 玩家回合 AIturn = 0 # 电脑回合 flag = Myturn # 设置轮流黑白手标志 game_state = 0 # 游戏状态, 0为未决胜负, 1为黑子胜, -1为白子胜 b_num = 0 # 黑子个数 w_num = 0 # 白子个数 MAX = 100000000000 # 正无穷 MIN = -100000000000 # 负无穷 deep = 3 # 规定ai搜索深度 drawed = True # 画出一方的棋另一方再走 point = [[-1] * 3 for i in range(200)] # 存储n中走法对应的点的横坐标[n][0]纵坐标[n][1]及value值[n][2] num_points = 0 # 记录可走步的个数 p1 = [0, 0] # p2 = [0, 0] # pygame.init() # 初始化pygame pygame.mixer.init() # 初始化mixer # 设置屏幕大小和标题 size = width, height = 740, 640 # 设置窗口大小 screen = pygame.display.set_mode(size) # 显示窗口 pygame.display.set_caption("欢乐五子棋 -- （HMR出品）") font = pygame.font.SysFont('comicsansms', 16) # 字体 ai_text = font.render('AI Steps', True, (84, 115, 135)) player_text = font.render('Player Steps', True, (84, 115, 135)) clock = pygame.time.Clock() # 设置时钟，定时刷新 whiteq = [] # 存储白棋位置 blackq = [] # 存储黑棋位置 chess = [[0] * 15 for i in range(15)] # 用来判断棋盘某处是否有棋,1为黑棋, -1为白棋, 0为空 # 更新chess状态 def renew_chess(chess, blackq, whiteq): for xx in range(15): for yy in range(15): chess[xx][yy] = 0 for xx, yy in blackq: chess[int((yy - 20) / 40)][int((xx - 20) / 40)] = 1 # 黑棋 for xx, yy in whiteq: chess[int((yy - 20) / 40)][int((xx - 20) / 40)] = -1 # 白棋 # 按钮响应函数 def do_start(btn): # 开始按钮的响应函数 global game_state, flag, b_num, w_num, blackq, whiteq game_state = 0 # 重置游戏状态及其他全局变量 b_num = 0 w_num = 0 blackq = [] whiteq = [] if first == me_first: flag = Myturn if first == ai_first: flag = AIturn def do_quit(btn): # 退出按钮的响应函数 pygame.quit() exit() def do_ai_first(btn): # 电脑先手 global first, flag first = ai_first do_start(btn) def do_me_first(btn): # 玩家先手 global first, flag first = me_first do_start(btn) # 画棋盘背景 def draw_background(screen): # 加载背景图片 bg_image = pygame.image.load('D:/py/5go/bg3.png') # 加载图片 screen.blit(bg_image, (20, 20)) # 画网格线，15x15 for line in r_lines: pygame.draw.line(screen, colors["black"], line[0], line[1], 2) for i in range(15): pygame.draw.line(screen, colors["black"], (20, 20 + 40 * i), (620, 20 + 40 * i), 1) # 画行 pygame.draw.line(screen, colors["black"], (20 + 40 * i, 20), (20 + 40 * i, 620), 1) # 画列 # 画棋子 def draw_gos(screen, blackq): # 画黑棋 for xq, yq in blackq: b_ = pygame.image.load('D:/py/5go/black.png') # 加载图片 b_rect = b_.get_rect() b_rect.centerx = xq b_rect.centery = yq screen.blit(b_, b_rect) # 画白棋 for xq, yq in whiteq: w_ = pygame.image.load('D:/py/5go/white.png') # 加载图片 w_rect = w_.get_rect() w_rect.centerx = xq w_rect.centery = yq screen.blit(w_, w_rect) # 获取以p点为中心的9个点的连线棋形 def getline(chess, px, py, dir_offset, discolor): line = [0 for i in range(9)] tmp_x = px + (-5 * dir_offset[0]) tmp_y = py + (-5 * dir_offset[1]) for i in range(9): tmp_x += dir_offset[0] tmp_y += dir_offset[1] if (tmp_x < 0 or tmp_x >= 15 or tmp_y < 0 or tmp_y >= 15): line[i] = discolor # 若超界则假设为对手棋 else: line[i] = chess[tmp_y][tmp_x] # 黑1白-1 return line # 分析棋形并得到该方向的分值,最后的参数是,有利分为1不利分为-1 def get_line_score(line, adcolor, discolor, weight, px, py): global l, values new_line = [[] for i in range(9)] num = 0 # 统计连子个数 for i in range(line.index(adcolor), 9): if line[i] == adcolor: num += 1 # 将line格式统一化,转字符串,己方棋子为1，对方棋子为2，空为0，便于匹配 for i in range(9): if line[i] == adcolor: new_line[i] = '1' elif line[i] == discolor: new_line[i] = '2' else: new_line[i] = '0' s = ''.join(new_line) # 转化为字符串 fan_s = s[::-1] # 字符串反转 # 逐个匹配 if num > 5: return values["11111"] * weight if 5 >= num >= 1: for each in l[num - 1]: result = re.search(each, s) if result != None: # 找到匹配项 return (values[result.group()] + values_of_pos[py][px]) * weight result = re.search(each, fan_s) if result != None: # 找到匹配项 return (values[result.group()] + values_of_pos[py][px]) * weight # 未找到匹配项，为单子情况 return (values["others"] + values_of_pos[py][px]) * weight # 评估函数，评估当前棋局的分值 def Evaluate(chess): advantage = 0 # 有利分, 对上一层有利的分！！ disadvantage = 0 # 不利分, 玩家所持棋子得分, 取负 adcolor = 0 # ai所持棋子颜色 discolor = 0 # 玩家所持棋子颜色 dir_offset = [(1, 0), (0, 1), (1, 1), (1, -1)] # 四个方向，右，上，右上，右下 if first == me_first: adcolor = 1 # ai持白子 discolor = -1 if first == ai_first: adcolor = -1 # ai持黑子 discolor = 1