算法题:2n皇后问题(Python)


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)

 

上一篇:蓝桥杯 DFS 2n皇后问题(困难)


下一篇:打印2n+1的数字菱形(SCAU期末考试题)