标签：mat Python 算法 range 放置 皇后 2n col mkB

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)

标签：mat,Python,算法,range,放置,皇后,2n,col,mkB
来源： https://blog.csdn.net/hoo_chsodene/article/details/114370833

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。