python-2n皇后问题

解题思路:

这道题其实和n皇后的题没有太大的差别。

做法是这样的，首先找出黑皇后的所有结果（这里暂时不考虑白皇后）。

详细做法可参考我之间写的一篇文章，介绍很详细。https://blog.dotcpp.com/a/83292。

那么单独考虑白皇后时结果和单独考虑黑皇后是一样的。

接下来写一个双重循环就好了。

找出黑皇后和白皇后在每一行摆放都不在同一位置的组合就OK了。

注意事项:

参考代码:

def dfs(A,cur):  
    global n  
    global Black  
  
    if cur == n:  
        Black.append(A[:])  #注意这里不要写成Black.append(A)，会出错，具体什么原因还不太清楚
        return  
    for j in range(n):  
        if Marix[cur][j]:  
            A[cur] = j  
            for i in range(cur):  
                if A[i] == j or abs(j - A[i]) == cur - i:  
                    break  
            else:  
                dfs(A,cur+1)  
      
if __name__ == '__main__':  
    n = int(input())  
    Marix = [[int(j) for j in input().strip().split()] for i in range(n)]  
    Black = []  
    dfs([None]*n,0)  
    White = Black[:]  
  
    ans = 0  
    for i in range(len(White)):  
        for j in range(len(Black)):  
            for k in range(n):      #找出黑皇后和白皇后在每一行摆放都不在同一位置的组合
               if White[i][k] == Black[j][k]:  
                    break  
            else:  
                ans = ans+1  
    print(ans)