python N皇后

python N皇后

问题描述

N皇后问题是8皇后问题的扩展，八皇后问题，是一个古老而著名的问题，是回溯算法的典型案例。该问题是国际西洋棋棋手马克斯·贝瑟尔于1848年提出：在8×8格的国际象棋上摆放八个皇后，使其不能互相攻击，即任意两个皇后都不能处于同一行、同一列或同一斜线上，求所有的摆法。如图为8皇后问题的一种摆法

基本思路

是一个很典型的dfs（深度优先搜索）搜索回溯问题

且dfs函数部分很简单，就是dfs深搜模板套上，主要是判断皇后摆放位置的判断问题

判断对角线上的值

左上每个表格xy值相加都为一个结果

而左下表格xy相减值也是一样的

而是否在一条线上遍历即可

代码实现

def check(x,y):
    for i in range(n):
        if nums[i][y] == 1:         #判断同一直线
            return False
    for i in range(n):
        for j in range(n):
            if x + y == i + j and nums[i][j] == 1:      #左上对角线
                return False
    for i in range(n):
        for j in range(n):
            if x - y == i - j and nums[i][j] == 1:      #左下对角线
                return False
    return True
def dfs(num):
    global result
    if num == n:        #如果布置的皇后刚好和输入相同
        result += 1
        return
    for i in range(n):      #行是由递进一层控制，只要遍历每列
        if check(num,i):
            nums[num][i] = 1        #标记
            dfs(num + 1)
            nums[num][i] = 0        #恢复，回溯


n = int(input())
nums = [[0]*n for _ in range(n)]
result = 0
dfs(0)      #从第 0 行开始
print(result)

输入

8

运行结果

92