问题描述
给定一个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实现)