蓝桥杯基础练习VIP-2n皇后问题（python解法）

题目描述

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

输入

输入的第一行为一个整数n，表示棋盘的大小。 n小于等于8

接下来n行，每行n个0或1的整数，如果一个整数为1，表示对应的位置可以放皇后，如果一个整数为0，表示对应的位置不可以放皇后。
输出
输出一个整数，表示总共有多少种放法。

样例输入

4
1 1 1 1 
1 1 1 1 
1 1 1 1 
1 1 1 1 

样例输出

2

答案

n=int(input())

l=[]#记录棋盘，全局共用
for i in range(n):
    temp=list(map(int,input().split()))
    l.append(temp)#读入数据到二维列表

#此处白黑各用三个
count=0#存放可能的总个数
col1=list(0 for x in range(n))#存放已经被用过的列
up1=list(0 for x in range(2*n-1))#存放用过的上对角线,行+列表示
down1=list(0 for x in range(2*n-1))#存放用过的下对角线，行-列表示

col2=list(0 for x in range(n))#存放已经被用过的列
up2=list(0 for x in range(2*n-1))#存放用过的上对角线,行+列表示
down2=list(0 for x in range(2*n-1))#存放用过的下对角线，行-列表示

def put_1(x):
    if(x==n):
    #达到目标状态，放好黑皇后，去放白皇后
        put_2(0)
    else:
        for j in range(n):
            #如果新状态合法，即此列/此对角线和、差未被使用
            if(col1[j]==0 and up1[x+j]==0 and down1[x-j]==0 and l[x][j]==1):
                col1[j]=1
                up1[x+j]=1
                down1[x-j]=1
                l[x][j]=0
                #再去下一行放置
                put_1(x+1)
                #每跳出一层，需要清除此层标记
                col1[j]=0
                up1[x+j]=0
                down1[x-j]=0
                l[x][j]=1

def put_2(x):
    global count
    if(x==n):
    #达到目标状态，计数＋1
        count+=1
    else:
    #未达到目标状态，对于每一个新状态，
        for j in range(n):
            #如果新状态合法，即此列/此对角线和、差未被使用
            if(col2[j]==0 and up2[x+j]==0 and down2[x-j]==0 and l[x][j]==1):
                col2[j]=1
                up2[x+j]=1
                down2[x-j]=1
                l[x][j]=0
                #再去下一行放置
                put_2(x+1)
                #每跳出一层，需要清除此层标记
                col2[j]=0
                up2[x+j]=0
                down2[x-j]=0
                l[x][j]=1

put_1(0)
print(count)