算法题：2n皇后问题（Python）

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

样例输入：

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

样例输出：

2

def checkblack(i, j):
    for k in range(j):
        # 行检查 or 对角线检查
        if mkB[k]==i or (j-k == abs(i-mkB[k])):
            return False
    return True

def checkwhite(i,j):
    #与黑皇后的冲突检查
    for k in range(n):
        if k==j and i==mkB[k]:
            return False
    #与白皇后的冲突检查
    for k in range(j):
        # 行检查 or 对角线检查
        if mkW[k]==i or (j-k == abs(i-mkW[k])):
            return False
    return True

def putwhite(col):
    global n
    global num
    if col == n+1:
        # 白皇后也放置完毕
        # 计数一次
        num += 1
    else:
        # 遍历各行
        for i in range(n):
            j = col-1
            # 检验是否能放置
            # 检验是否占用黑皇后的位置
            # 检验是否与其他白皇后在行或对角线上冲突
            if mat[i][j]!=0 and checkwhite(i,j):
                # 可以放置
                # 记录位置
                mkW[j] = i
                # 在下一列放置白皇后
                putwhite(col+1)
            else:
                continue

def putblack(col):
    global n
    if col==n+1:
        # 已放好n个黑皇后
        # 开始放置白皇后
        putwhite(1)
    else:
        # 遍历各行
        for i in range(n):
            j = col-1
            # 检验是否能放置
            # 检查是否与其他黑皇后在行或对角线上冲突
            if mat[i][j] != 0 and checkblack(i, j):
                # 可以放置
                # 记录位置
                mkB[j]=i
                # 在下一列放置黑皇后
                putblack(col+1)
            else:
                continue

# 2n皇后
n = eval(input())
mat = list()
for i in range(n):
    mat.append(input().split(' '))
for i in range(n):
    mat[i] = list( map(eval, mat[i]) )
mkB = [0]*n
mkW = [0]*n
num = 0
putblack(1)
print(num)