from graphics import *
from math import *
import numpy as np
def ai():
"""
AI計算落子位置
"""
maxmin(True, DEPTH, -99999999, 99999999)
return next_point[0], next_point[1]
def maxmin(is_ai, depth, alpha, beta):
"""
負值極大算法搜索 alpha + beta剪枝
"""
# 遊戲是否結束 | | 探索的遞歸深度是否到邊界
if game_win(list1) or game_win(list2) or depth == 0:
return evaluation(is_ai)
blank_list = list(set(list_all).difference(set(list3)))
order(blank_list) # 搜索順序排序 提升剪枝效率
# 遍歷每個候選步
for next_step in blank_list[0:60]:
# 若是要評估的位置沒有相鄰的子， 則不去評估 減小計算
if not has_neightnor(next_step):
continue
if is_ai:
list1.append(next_step)
else:
list2.append(next_step)
list3.append(next_step)
value = -maxmin(not is_ai, depth - 1, -beta, -alpha)
if is_ai:
list1.remove(next_step)
else:
list2.remove(next_step)
list3.remove(next_step)
if value > alpha:
if depth == DEPTH:
next_point[0] = next_step[0]
next_point[1] = next_step[1]
# alpha + beta剪枝點
if value >= beta:
return beta
alpha = value
return alpha
def order(blank_list):
"""
離最後落子的鄰居位置最有多是最優勢
計算最後落子點的8個方向鄰居節點
若未落子，則插入到blank列表的最前端
:param blank_list: 未落子節點集合
:return: blank_list
"""
last_pt = list3[-1]
# for item in blank_list:
for i in range(-1, 2):
for j in range(-1, 2):
if i == 0 and j == 0:
continue
if (last_pt[0] + i, last_pt[1] + j) in blank_list:
blank_list.remove((last_pt[0] + i, last_pt[1] + j))
blank_list.insert(0, (last_pt[0] + i, last_pt[1] + j))
def has_neightnor(pt):
"""
判斷是否有鄰居節點
:param pt: 待評測節點
:return:
"""
for i in range(-1, 2):
for j in range(-1, 2):
if i == 0 and j == 0:
continue
if (pt[0] + i, pt[1] + j) in list3:
return True
return False
def evaluation(is_ai):
"""
評估函數
"""
if is_ai:
my_list = list1
enemy_list = list2
else:
my_list = list2
enemy_list = list1
# 算本身的得分
score_all_arr = [] # 得分形狀的位置 用於計算若是有相交 得分翻倍
my_score = 0
for pt in my_list:
m = pt[0]
n = pt[1]
my_score += cal_score(m, n, 0, 1, enemy_list, my_list, score_all_arr)
my_score += cal_score(m, n, 1, 0, enemy_list, my_list, score_all_arr)
my_score += cal_score(m, n, 1, 1, enemy_list, my_list, score_all_arr)
my_score += cal_score(m, n, -1, 1, enemy_list, my_list, score_all_arr)
# 算敵人的得分， 並減去
score_all_arr_enemy = []
enemy_score = 0
for pt in enemy_list:
m = pt[0]
n = pt[1]
enemy_score += cal_score(m, n, 0, 1, my_list, enemy_list, score_all_arr_enemy)
enemy_score += cal_score(m, n, 1, 0, my_list, enemy_list, score_all_arr_enemy)
enemy_score += cal_score(m, n, 1, 1, my_list, enemy_list, score_all_arr_enemy)
enemy_score += cal_score(m, n, -1, 1, my_list, enemy_list, score_all_arr_enemy)
total_score = my_score - enemy_score * 0.1
return total_score
def cal_score(m, n, x_decrict, y_derice, enemy_list, my_list, score_all_arr):
"""
每一個方向上的分值計算
:param m:
:param n:
:param x_decrict:
:param y_derice:
:param enemy_list:
:param my_list:
:param score_all_arr:
:return:
"""
add_score = 0 # 加分項
# 在一個方向上， 只取最大的得分項
max_score_shape = (0, None)
# 若是此方向上，該點已經有得分形狀，不重複計算
for item in score_all_arr:
for pt in item[1]:
if m == pt[0] and n == pt[1] and x_decrict == item[2][0] and y_derice == item[2][1]:
return 0
# 在落子點 左右方向上循環查找得分形狀
for offset in range(-5, 1):
# offset = -2
pos = []
for i in range(0, 6):
if (m + (i + offset) * x_decrict, n + (i + offset) * y_derice) in enemy_list:
pos.append(2)
elif (m + (i + offset) * x_decrict, n + (i + offset) * y_derice) in my_list:
pos.append(1)
else:
pos.append(0)
tmp_shap5 = (pos[0], pos[1], pos[2], pos[3], pos[4])
tmp_shap6 = (pos[0], pos[1], pos[2], pos[3], pos[4], pos[5])
for (score, shape) in shape_score:
if tmp_shap5 == shape or tmp_shap6 == shape:
if score > max_score_shape[0]:
max_score_shape = (score, ((m + (0 + offset) * x_decrict, n + (0 + offset) * y_derice),
(m + (1 + offset) * x_decrict, n + (1 + offset) * y_derice),
(m + (2 + offset) * x_decrict, n + (2 + offset) * y_derice),
(m + (3 + offset) * x_decrict, n + (3 + offset) * y_derice),
(m + (4 + offset) * x_decrict, n + (4 + offset) * y_derice)),
(x_decrict, y_derice))
# 計算兩個形狀相交， 如兩個3活 相交， 得分增長 一個子的除外
if max_score_shape[1] is not None:
for item in score_all_arr:
for pt1 in item[1]:
for pt2 in max_score_shape[1]:
if pt1 == pt2 and max_score_shape[0] > 10 and item[0] > 10:
add_score += item[0] + max_score_shape[0]
score_all_arr.append(max_score_shape)
return add_score + max_score_shape[0]
def game_win(list):
"""
勝利條件判斷
"""
# for m in range(COLUMN):
# for n in range(ROW):
# if n < ROW - 4 and (m, n) in list and (m, n + 1) in list and (m, n + 2) in list and (
# m, n + 3) in list and (m, n + 4) in list:
# return True
# elif m < ROW - 4 and (m, n) in list and (m + 1, n) in list and (m + 2, n) in list and (
# m + 3, n) in list and (m + 4, n) in list:
# return True
# elif m < ROW - 4 and n < ROW - 4 and (m, n) in list and (m + 1, n + 1) in list and (
# m + 2, n + 2) in list and (m + 3, n + 3) in list and (m + 4, n + 4) in list:
# return True
# elif m < ROW - 4 and n > 3 and (m, n) in list and (m + 1, n - 1) in list and (
# m + 2, n - 2) in list and (m + 3, n - 3) in list and (m + 4, n - 4) in list:
# return True
return False
def draw_window():
"""
繪製棋盤
"""
# 繪製畫板
win = GraphWin("五子棋", GRAPH_HEIGHT, GRAPH_WIDTH)
win.setBackground("gray")
# 繪製列
i1 = 0
while i1 <= GRID_WIDTH * COLUMN:
i1 = i1 + GRID_WIDTH
l = Line(Point(i1, GRID_WIDTH), Point(i1, GRID_WIDTH * COLUMN))
l.draw(win)
# 繪製行
i2 = 0
while i2 <= GRID_WIDTH * ROW:
i2 = i2 + GRID_WIDTH
l = Line(Point(GRID_WIDTH, i2), Point(GRID_WIDTH * ROW, i2))
l.draw(win)
return win
def main():
"""
程序循環
:return:
"""
mode = int(input("先手 AI先手 ? 1 0 \n"))
# 繪製棋盤
win = draw_window()
# 添加棋盤全部點
for i in range(COLUMN + 1):
for j in range(ROW + 1):
list_all.append((i, j))
# 循環條件
g = 0
change = 0
# 開始循環
while g == 0:
# AI
if change % 2 == mode:
# AI先手 走天元
if change == 0:
pos = (6, 6)
else:
pos = ai()
# 添加落子
list1.append(pos)
list3.append(pos)
# 繪製白棋
piece = Circle(Point(GRID_WIDTH * (pos[0]), GRID_WIDTH * (pos[1])), 12)
piece.setFill('white')
piece.draw(win)
# AI勝利
if game_win(list1):
message = Text(Point(GRAPH_WIDTH / 2, GRID_WIDTH / 2), "AI獲勝")
message.draw(win)
g = 1
change = change + 1
# User
else:
p2 = win.getMouse()
x = round((p2.getX()) / GRID_WIDTH)
y = round((p2.getY()) / GRID_WIDTH)
# 若點未被選取過
if not (x, y) in list3:
# 添加落子
list2.append((x, y))
list3.append((x, y))
# 繪製黑棋
piece = Circle(Point(GRID_WIDTH * x, GRID_WIDTH * y), 12)
piece.setFill('black')
piece.draw(win)
# 勝利
if game_win(list2):
message = Text(Point(GRAPH_WIDTH / 2, GRID_WIDTH / 2), "人類勝利")
message.draw(win)
g = 1
change = change + 1
message = Text(Point(GRAPH_WIDTH / 2 + 100, GRID_WIDTH / 2), "遊戲結束")
message.draw(win)
win.getMouse()
win.close()
if __name__ == '__main__':
GRID_WIDTH = 40
COLUMN = 11
ROW = 11
GRAPH_WIDTH = GRID_WIDTH * (ROW + 1)
GRAPH_HEIGHT = GRID_WIDTH * (COLUMN + 1)
list1 = [] # AI
list2 = [] # human
list3 = [] # all
list_all = [] # 整個棋盤的點
next_point = [0, 0] # AI下一步最應該下的位置
mode=int(input("請選擇: 快不許 或 慢卻準 ? 1 : 0 \n"))
if mode==1:
DEPTH=1
elif mode==0:
DEPTH=3
else:
DEPTH=3
shape_score = [(50, (0, 1, 1, 0, 0)),
(50, (0, 0, 1, 1, 0)),
(200, (1, 1, 0, 1, 0)),
(500, (0, 0, 1, 1, 1)),
(500, (1, 1, 1, 0, 0)),
(5000, (0, 1, 1, 1, 0)),
(5000, (0, 1, 0, 1, 1, 0)),
(5000, (0, 1, 1, 0, 1, 0)),
(5000, (1, 1, 1, 0, 1)),
(5000, (1, 1, 0, 1, 1)),
(5000, (1, 0, 1, 1, 1)),
(5000, (1, 1, 1, 1, 0)),
(5000, (0, 1, 1, 1, 1)),
(50000, (0, 1, 1, 1, 1, 0)),
(99999999, (1, 1, 1, 1, 1))]
main()