蓝桥杯练习系统2n皇后问题（Python）

问题描述

给定一个n*n的棋盘，棋盘中有一些位置不能放皇后。现在要向棋盘中放入n个黑皇后和n个白皇后，使任意的两个黑皇后都不在同一行、同一列或同一条对角线上，任意的两个白皇后都不在同一行、同一列或同一条对角线上。问总共有多少种放法？n小于等于8。

输入格式

输入的第一行为一个整数n，表示棋盘的大小。
　　接下来n行，每行n个0或1的整数，如果一个整数为1，表示对应的位置可以放皇后，如果一个整数为0，表示对应的位置不可以放皇后。

输出格式

输出一个整数，表示总共有多少种放法。

算法思路

首先是n行n列的输入
首先需要定义一个判断该位置是否可以放皇后的函数，输入参数为当前行号和列号以及皇后颜色，考虑一行一行的判断，所以是

def 判断该位置是否可以放皇后:
for 每一行：
	  if 同一列已经方皇后 或者 对角线已经放皇后：
	     return 

接下来需要定义一个检索黑皇后的函数，输入参数为行号

if 行号是否超过总行号：
	去判断白皇后
else 
   for 每一列：
      if 棋盘该位置是0表示不能放皇后
         结束本次循环
      elif 该位置是否可以放皇后：
         标记该位置
         进入下一次检索黑皇后的函数，行号需要+1
         递归结束后撤销对该位置的标记，毕竟后面白皇后还要检索呢

还需要定义一个检索白皇后的函数

if 行号是否超过总行号：
	放法+1
else 
   for 每一列：
      if 棋盘该位置是0表示不能放皇后或者该位置已经放了黑皇后
         结束本次循环
      elif 该位置是否可以放皇后：
         标记该位置
         进入下一次检索黑皇后的函数，行号需要+1
         递归结束后撤销对该位置的标记

参考代码

# 方法一
def is_can_put(i, j, color):
    flag = True
    for k in range(i):
        # 判断在↓同一列是否已经放置皇后
        # 判断在↖左上对角线是否已经放置皇后(and前面的保证序号不越界)
        # 判断在↗右上对角线是否已经放置皇后(and前面的保证序号不越界)
        if qipan[k][j] == color or\
                j-i+k >= 0 and qipan[k][j-i+k] == color or\
                i+j-k < n and qipan[k][i+j-k] == color:
            flag = False
            break
    return flag


def dfs_white(i):
    # 白皇后所有行都遍历完以后，开始判断黑皇后
    if i >= n:
        dfs_black(0)
    else:
        for j in range(n):
            # 如果该位置是0，结束本次循环
            if qipan[i][j] == 0:
                continue
            # 如果该位置是1，判断能不能放白皇后
            elif is_can_put(i, j, white):
                # 标记此处能放白皇后
                qipan[i][j] = white
                # 进入下一次循环
                dfs_white(i + 1)
                # 取消标记
                qipan[i][j] = 1

def dfs_black(i):
    global ans
    # 白皇后所有行都遍历完以后，开始判断黑皇后
    if i >= n:
        ans += 1
    else:
        for j in range(n):
            # 如果该位置是0或者该位置已经放了白皇后，结束本次循环
            if qipan[i][j] == 0 or qipan[i][j] == white:
                continue
            # 如果该位置是1，判断能不能放黑皇后
            elif is_can_put(i, j, black):
                # 标记此处能放黑皇后
                qipan[i][j] = black
                # 进入下一次循环
                dfs_black(i + 1)
                # 取消标记
                qipan[i][j] = 1


black = 3
white = 2
n = int(input())
qipan = []
for i in range(n):
    temp = list(map(int,input().split()))
    qipan.append(temp)
# print(qipan)
ans = 0
dfs_white(0)
print(ans)

一点感悟

理不清别人代码思路的时候，多使用单步调试工具

分享

这两篇博客都写的很好
Python——(蓝桥杯)2n皇后问题
N皇后问题（Python实现）