OR-Tools求解N皇后问题实例

1.N皇后问题

n 皇后问题研究的是如何将 n 个皇后放置在 n×n 的棋盘上，并且使皇后彼此之间不能相互攻击。

2.分析

N皇后问题一般都是采取搜索的方式求解，如BFS，DFS的方式进行求解。但我们现在使用Google的开源工具OR-Tools对这个问题进行建模，实现约束求解。

N皇后问题是一个的无目标的约束问题，所以只需要建立如下约束即可：

1. 任意两个皇后不能在同一行
2. 任意两个皇后不能在同一列
3. 任意两个皇后不能在一个对角线上

在实现对角线约束的时候分为两种情况：

1. 从左下往右上的对角线，这条对角线上格子的坐标的y - x的值相同。
2. 从右下往左上的对角线，这条对角线上格子的坐标的x + y的值相同。

利用以上两种特性进行对角线的约束。

3.实现代码

from ortools.sat.python import cp_model

N = 8

model = cp_model.CpModel()
# 已经保证了不会出现同一行
Q = [model.NewIntVar(0, N - 1, name='queen') for i in range(N)]
# 约束不在同一列
model.AddAllDifferent(Q)

# 对角线约束
diag1 = []
diag2 = []
for j in range(N):
    d1 = model.NewIntVar(0, 2 * N - 1, name='youshang')
    model.Add(d1 == Q[j] + j)
    d2 = model.NewIntVar(-N, N, name='youxia')
    model.Add(d2 == Q[j] - j)
    diag1.append(d1)
    diag2.append(d2)
model.AddAllDifferent(diag1)
model.AddAllDifferent(diag2)

solver = cp_model.CpSolver()
solver.Solve(model)

# 输出结果
for q in Q:
    print(solver.Value(q))

4.输出结果

输出的结果是每一行棋盘放置皇后的位置的列坐标。

4
6
1
5
2
0
3
7