1460: 蓝桥杯基础练习VIP-2n皇后问题（Python3）11行简洁方法

解题思路:
    子问题：

        1.求出一次n皇后的解；

        2.两次遍历这个解，剔除重复摆放棋子的情况和在棋盘不允许摆子的地方摆子的情况。

注意事项:
    利用标准库的全排列列出所有情况用于找出n皇后的解。

参考代码:

import itertools

n, cnt = int(input()), 0
map_ = [list(map(int, input().split())) for _ in range(n)]
b = list(itertools.permutations(i for i in range(n)))
b = list(filter(lambda x: n == len(set(x[i] - i for i in range(n))) == len(set(x[i] + i for i in range(n))), b))
for k in range(len(b)):
    for m in range(len(b)):
        if sum(1 for i, j in zip(b[k], b[m]) if i == j) == 0:  # check overlapping
            cnt += 1 if sum(1 for i, j in zip(range(n), b[k]) if map_[i][j] == 0) + \
                sum(1 for i_1, j_1 in zip(range(n), b[m]) if map_[i_1][j_1] == 0) == 0 else 0  # check legal placement
print(cnt)