从零构建五子棋AI:C++实现中的博弈树搜索与剪枝优化

张开发
2026/4/13 14:45:47 15 分钟阅读

分享文章

从零构建五子棋AI:C++实现中的博弈树搜索与剪枝优化
1. 五子棋AI的基本原理五子棋是一种双人策略型棋类游戏玩家通过在棋盘上交替落子先形成五子连线的一方获胜。要让计算机学会下五子棋我们需要教会它如何评估棋局并做出最佳决策。这就像教一个小朋友下棋只不过我们用的是代码和算法。博弈树是五子棋AI的核心数据结构。想象一下每次落子后棋盘都会出现新的局面这些局面就像树的分支一样不断展开。每个节点代表一个棋局状态子节点代表可能的下一步走法。对于15×15的标准棋盘每一步都有上百种可能的落子位置这使得博弈树会快速膨胀。在实际编程中我们会遇到两个关键问题一是如何高效地遍历这棵巨大的博弈树二是如何准确评估每个棋局的好坏。这就引出了极大极小算法和α-β剪枝技术。极大极小算法让AI像人类一样思考假设对手会采取最优策略选择对自己最有利的走法。而α-β剪枝则像是一个聪明的棋手能够快速排除明显不利的选择大幅减少需要评估的局面数量。2. 构建博弈树的数据结构2.1 Node类的设计在C实现中我们首先需要设计Node类来表示博弈树的节点。这个类需要包含以下核心信息class Node { public: int32_t value; // 节点估值 uint32_t depth; // 节点深度 uint8_t cntX, cntY; // 最后落子坐标 Node* father; // 父节点指针 setNode* children; // 子节点集合 uint8_t board[15][15]; // 棋盘状态 // 构造函数和其他成员函数... };每个节点都保存了完整的棋盘状态这样我们可以随时回溯到任意棋局。board数组用B表示黑子W表示白子0表示空位。depth记录节点在博弈树中的深度根节点深度为0每深入一层深度加1。2.2 棋局评估函数估价函数是AI的棋感它决定了AI如何判断局面的好坏。我们采用基于五元组的评估方法int32_t evaluate_black(string s) { // 黑子五元组评分规则 if (s BBBBB) return 1000000; if (s BBBB0) return 10000; // 其他模式... } int32_t evaluate_white(string s) { // 白子五元组评分规则 if (s WWWWW) return -10000000; if (s 0WWW0) return -2000; // 其他模式... }对于每个五元组横向、纵向或对角线方向的连续五个位置我们根据其中黑白子的分布给出评分。例如五个黑子连珠得100万分而四个黑子加一个空位得1万分。白子的威胁会被赋予更高的负分这体现了防守优先的策略。3. 实现极大极小搜索算法3.1 基本搜索流程极大极小算法的核心思想是交替考虑自己和对手的最佳走法。在实现中我们使用深度优先搜索遍历博弈树void GameTree::game() { // 检查是否已分胜负 uint8_t result nodeRoot-board_identify(); if (result ! 0) return; openTable.push_back(nodeRoot); while (!openTable.empty()) { Node* node openTable.front(); openTable.pop_front(); // 剪枝判断 if (is_alpha_beta_cut(node-father)) continue; // 扩展子节点 if (node-depth maxDepth) { if (expand_children_nodes(node) 0) continue; } // 评估叶节点 node-evaluate(); update_value_from_node(node); } // 选择最佳走法 set_next_pos(); }MAX节点AI回合会选择子节点中估值最高的走法而MIN节点对手回合会选择对AI最不利的走法。这种交替选择形成了我最好的情况下对手最差反应的决策模式。3.2 搜索优化技巧单纯的极大极小搜索效率很低我们需要一些优化技巧限制搜索深度通常设置4-8层平衡速度和棋力搜索半径只考虑已有棋子周围2-3格内的空位走法排序优先评估看起来更有希望的走法在代码中我们通过expandRadius参数控制搜索范围vectorpairuint8_t, uint8_t get_search_nodes(Node* node) { bool newBoard[15][15] {false}; // 只在已有棋子周围expandRadius范围内搜索 for (int i 0; i 15; i) { for (int j 0; j 15; j) { if (node-board[i][j] 0) continue; // 计算搜索范围... } } // 返回候选位置列表 }4. α-β剪枝的实现与优化4.1 剪枝原理α-β剪枝的核心思想是当发现某条路径不可能比已知的最佳选择更好时就停止继续搜索这条路径。这就像下棋时如果发现某步棋明显会导致劣势就不需要再考虑这步棋之后的各种变化了。在代码中我们维护两个值α值MAX节点至少能得到的分数β值MIN节点至多能让MAX得到的分数当某个节点的α ≥ β时就可以剪枝bool is_alpha_beta_cut(Node* node) { if (node nullptr) return false; if (node-is_max_node() node-value node-father-value) return true; if (!node-is_max_node() node-value node-father-value) return true; return is_alpha_beta_cut(node-father); }4.2 剪枝效果示例假设我们有以下博弈树片段MAX节点A当前α5MIN节点B是A的子节点正在评估B的子节点CMAX节点如果发现C的某个子节点D的估值7由于B是MIN节点它会选择子节点中的最小值。而A已经有一个α5B如果能找到一个≤5的值A就不会选择B这条路径。因此当我们发现D7时可以立即停止评估C的其他子节点因为B的最终值至少是7C的其他子节点可能更大而7已经大于A的α5所以A不会选择B这条路径。这种剪枝可以节省大量计算时间在最优情况下能将搜索深度增加一倍。5. 完整对弈流程的实现5.1 人机对弈模式我们通过GameTree类提供完整的人机对弈功能void machine_human_play() { uint8_t board[15][15] {0}; for (int k 0; k 225; k) { GameTree gt(9, 2, board); uint8_t result gt.game(); // 显示AI走法 gt.show_board(false); auto pos gt.get_next_pos(); board[pos.first][pos.second] B; // 获取玩家走法 int x, y; cin x y; board[x][y] W; } }AI会先计算并显示自己的走法然后等待玩家输入坐标。这个过程会一直持续直到有一方获胜或棋盘填满。5.2 机机对弈模式机机对弈模式可以让我们观察AI的自我对弈void machine_machine_play() { uint8_t board[15][15] {0}; uint8_t turn B; while (true) { GameTree gt(8, 2, board); uint8_t result gt.game(); auto pos gt.get_next_pos(); board[pos.first][pos.second] turn; turn (turn B) ? W : B; // 显示当前棋局... } }这个模式对于调试AI策略特别有用可以观察AI在攻防转换中的决策过程。6. 性能优化与进阶技巧6.1 估价函数改进基础的估价函数虽然有效但仍有改进空间增加特殊棋型的识别如活四、冲四、活三等考虑棋型的组合效应多个威胁的组合应该获得更高评分引入位置权重中心位置通常更有价值例如可以改进evalute函数void Node::evaluate() { value 0; // 基础五元组评估... // 增加位置权重 static int positionWeight[15][15] { /*...*/ }; for (int i 0; i 15; i) { for (int j 0; j 15; j) { if (board[i][j] B) value positionWeight[i][j]; if (board[i][j] W) value - positionWeight[i][j]; } } // 特殊棋型检测... }6.2 并行化搜索博弈树的搜索天然适合并行化。我们可以使用多线程同时评估不同分支使用异步更新α-β值实现迭代深化搜索逐步增加搜索深度C的库可以方便地实现多线程void parallel_search(Node* node) { vectorthread threads; for (Node* child : node-children) { threads.emplace_back([child](){ GameTree gt; gt.nodeRoot child; gt.game(); }); } for (auto t : threads) t.join(); }7. 常见问题与调试技巧在开发五子棋AI过程中我遇到过几个典型问题AI忽视明显威胁这通常是因为估价函数对某些棋型评分不足。解决方法是增加对应棋型的检测和评分。搜索速度过慢可以通过以下方式优化减小搜索半径优化数据结构比如使用位棋盘表示实现更高效的剪枝重复走法引入随机因素当多个走法评分相近时随机选择一个。调试时可以输出搜索树和估值信息void debug_print(Node* node) { cout Depth: node-depth Value: node-value endl; // 打印棋盘... for (Node* child : node-children) { debug_print(child); } }8. 从五子棋AI到其他棋类这套博弈树框架可以扩展到其他棋类游戏主要需要调整棋盘表示和走法生成胜负判断逻辑估价函数的设计例如实现围棋AI时需要更复杂的棋盘表示气、眼等概念蒙特卡洛树搜索等更高级的算法考虑征子、打劫等特殊规则但核心的极大极小思想和α-β剪枝仍然适用。

更多文章