蓝桥 基础练习 2n皇后问题 python

问题描述

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

输入格式

　　输入的第一行为一个整数n，表示棋盘的大小。
　　接下来n行，每行n个0或1的整数，如果一个整数为1，表示对应的位置可以放皇后，如果一个整数为0，表示对应的位置不可以放皇后。

输出格式

　　输出一个整数，表示总共有多少种放法。

样例输入

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

样例输出

2

样例输入

4
1 0 1 1
1 1 1 1
1 1 1 1
1 1 1 1

样例输出

0

解题思路：使用深度搜索dfs

def dfs(nums,h,colour):
    if colour==3 and h==n:
        global count
        count += 1
    elif colour == 2 and h == n:
        dfs(nums,0,3)
    else:
        for i in range(n):
            if text(nums,h,i,colour):
                nums[h][i] = colour
                dfs(nums,h+1,colour)
                nums[h][i] = 1

def text(nums,h,i,colour):
    if nums[h][i] != 1:
        return 0
    else:
        for a in range(h):
            for b in range(n):
                if nums[a][b] == colour:
                    if b == i:
                        return 0
                    if abs(a-h) == abs(b-i):
                        return 0
        return 1

n = int(input())
nums = []
count = 0
for i in range(n):
    num = list(map(int,input().split()))
    nums.append(num)
dfs(nums,0,2)
print(count)

python知识点：

global可以在定义的函数中使用全局变量并且改变全局变量