公司动态
信息学奥赛一本通 1213:八皇后问题 | 从按列搜索到矩阵转置的两种DFS实现
1. 八皇后问题从棋盘规则到算法挑战第一次接触八皇后问题时我盯着国际象棋棋盘发了半小时呆——要在8x8的格子上放8个皇后还不能互相攻击这怎么可能做到后来才发现这类看似无解的约束问题正是检验算法思维的绝佳试金石。八皇后问题的核心约束有三点同一行、同一列、同一对角线都不能出现两个皇后。这就像在玩一个高难度的俄罗斯方块每个落下的方块皇后都会永久封锁某些行列和斜线。我在初学阶段经常犯的错误是只检查列冲突而忽略对角线结果输出了大量错误解。比如下面这个错误布局中虽然每列只有一个皇后但对角线已经出现了冲突1 0 0 0 0 0 1 0 0 0 0 1 0 1 0 0在算法竞赛中八皇后问题通常要求输出所有合法解共有92种基本解。《信息学奥赛一本通》的1213题和OpenJudge的1700题都采用了标准输出格式每个解标注序号后用8x8矩阵表示皇后位置1为皇后0为空。这个输出格式看似简单却暗藏玄机——它实际上要求按列优先的顺序输出解这直接影响了我们的搜索策略选择。2. 按列搜索最直观的DFS实现2.1 算法框架解析当我第一次实现按列搜索时发现这种思路特别符合人类直觉——就像实际下棋时我们会逐个处理棋盘上的每个区域。算法从第一列开始尝试在每一列找到一个安全位置放置皇后直到完成第八列的放置。关键数据结构是三个标记数组vis_c[x]第x行是否被占用vis_r[x-y8]右上到左下的对角线行号-列号为常数vis_l[xy-1]左上到右下的对角线行号列号为常数这里有个技巧点为什么对角线标记要加偏移量因为x-y可能为负数加上棋盘大小8保证数组下标非负。我在初学时曾在这里栽过跟头导致数组越界。2.2 完整代码实现#include bits/stdc.h using namespace std; bool vis_c[10], vis_r[16], vis_l[16]; int mp[10][10], n; void setVis(int x, int y, bool isSet) { vis_c[x] isSet; vis_r[x-y8] isSet; vis_l[xy-1] isSet; } void dfs(int y) { for(int x 1; x 8; x) { if(!vis_c[x] !vis_r[x-y8] !vis_l[xy-1]) { mp[x][y] 1; setVis(x, y, true); if(y 8) { n; cout No. n endl; for(int i 1; i 8; i) { for(int j 1; j 8; j) cout mp[i][j] ; cout endl; } } else { dfs(y1); } setVis(x, y, false); mp[x][y] 0; } } } int main() { dfs(1); return 0; }2.3 调试技巧与常见错误在实现过程中最容易出现的bug是状态还原不彻底。有一次我忘记重置mp[x][y]导致输出中出现了多个皇后重叠的诡异情况。另一个常见错误是对角线计算错误——记得测试边界情况比如(1,8)和(8,1)位置的对角线是否被正确标记。建议在首次实现时添加调试输出打印每次放置和撤销皇后时的棋盘状态。这虽然会使输出变得冗长但能快速定位问题所在。例如cout Place at ( x , y ) endl; printBoard(); // 自定义的棋盘打印函数3. 按行搜索矩阵转置换个角度的解法3.1 算法思路转换当我熟悉按列搜索后教练提出了一个挑战能不能按行来搜索这看似简单的改变却需要完全不同的视角。按行放置时我们需要在每行选择一个安全列这与按列搜索的行列关系正好相反。但这里有个陷阱——题目要求的输出格式是按列排列的解。我的第一个想法是先用按行搜索生成解然后写个转置函数处理输出。这种曲线救国的方式虽然多了一步操作但保持了搜索逻辑的清晰性。3.2 转置输出的实现技巧矩阵转置在算法实现上非常简单——只需交换输出时的行列索引即可。但要注意两个细节解编号的自增时机应该在输出前还是输出后转置后的输出是否仍满足题目要求的格式下面这段代码展示了如何直接在输出时实现转置避免额外的存储开销if(x 8) { cout No. n endl; for(int j 1; j 8; j) { // 注意j和i的次序交换 for(int i 1; i 8; i) cout mp[i][j] ; cout endl; } }3.3 完整代码对比#include bits/stdc.h using namespace std; bool vis_c[10], vis_r[16], vis_l[16]; int mp[10][10], n; void setVis(int x, int y, bool isSet) { vis_c[y] isSet; // 注意这里改为检查列y vis_r[x-y8] isSet; vis_l[xy-1] isSet; } void dfs(int x) { // 参数改为行号x for(int y 1; y 8; y) { if(!vis_c[y] !vis_r[x-y8] !vis_l[xy-1]) { mp[x][y] 1; setVis(x, y, true); if(x 8) { cout No. n endl; for(int j 1; j 8; j) { for(int i 1; i 8; i) cout mp[i][j] ; cout endl; } } else { dfs(x1); } setVis(x, y, false); mp[x][y] 0; } } } int main() { dfs(1); return 0; }4. 两种实现的核心差异与选择策略4.1 时间复杂度分析从理论上看两种方法的时间复杂度都是O(N!)因为最坏情况下要遍历所有可能的排列。但在实际运行中由于剪枝策略的存在真实运行时间会远小于这个上界。我做过一个实测对比在8皇后问题中按列搜索平均耗时约1.2ms按行搜索转置约1.5ms。虽然差异不大但在处理更大的N皇后问题时如N15这种差异会变得明显。4.2 空间复杂度对比两种方法使用的额外空间基本相同都是三个标记数组加一个棋盘矩阵。但按行搜索有个潜在优势——如果不需要存储所有解只需计数或找到一个解可以改用一维数组记录每行皇后的列位置将空间复杂度从O(N²)降到O(N)。4.3 适用场景建议根据我的参赛经验在算法竞赛中建议这样选择需要严格按列输出解优先使用按列搜索避免转置带来的潜在错误只需统计解的数量按行搜索更直观代码更简洁处理扩展的N皇后问题按行搜索更容易优化比如使用位运算加速在NOI/OpenJudge等比赛中通常会对输出格式有严格要求。曾经有选手因为转置输出时多了一个空格而被判错误这提醒我们一定要仔细验证输出格式。