洛谷 P1219八皇后 Python

题目描述

一个如下的 6×6 的跳棋棋盘，有六个棋子被放置在棋盘上，使得每行、每列有且只有一个，每条对角线（包括两条主对角线的所有平行线）上至多有一个棋子。

上面的布局可以用序列 2 4 6 1 3 5来描述，第 i 个数字表示在第 i 行的相应位置有一个棋子，如下：

行号 1 2 3 4 5 6

列号 2 4 6 1 3 5

这只是棋子放置的一个解。请编一个程序找出所有棋子放置的解。
并把它们以上面的序列方法输出，解按字典顺序排列。
请输出前 3 个解。最后一行是解的总个数。

输入格式

一行一个正整数 n，表示棋盘是 n×n 大小的。

输出格式

前三行为前三个解，每个解的两个数字之间用一个空格隔开。第四行只有一个数字，表示解的总数。

题目解析

一行一个正整数 n，表示棋盘是 n×n 大小的。

这题还是有些复杂的，典型的深度优先，不想写题解了呀，以后想写再来写啦~

# 输入
n = int(input())

num = [0 for i in range(n+1)]    # 记录需要保存的行号
vis = [0 for i in range(n+1)]    # 记录这个位置是否访问过
sumP = [0 for i in range(2 * n + 1)]    # 记录所经过的位置的横纵坐标之和
subP = [0 for i in range(2 * n + 1)]    # 记录所经过的位置的横纵坐标之差

count = 0    # 记录可以实现的方案之和

def DFS(step):
    # 结束条件
    if step == n+1:
        global count
        count += 1
        # 打印前三行
        if count <= 3:
            for i in range(1, n+1):
                print(num[i], end=" ")
            print()


    # 找step当前位置是否有合适的数
    for i in range(1,n+1):
        if vis[i] == 0:
            # 当前位置Check: [i, step]
            # 如果当前位置的对角线上没有元素
            if sumP[i + step] == 0 and subP[i - step + n] == 0:
                num[step] = i    # 记录当前元素
                vis[i] = 1       # 当前元素访问过，记录下来
                sumP[i + step] = 1    # 记录当前位置的横纵坐标之和
                subP[i - step + n] = 1    # 记录当前位置的横纵坐标之差，考虑会有负数，加上n
                DFS(step+1)
                vis[i] = 0    # 回溯，去除访问标记
                sumP[i + step] = 0
                subP[i - step + n] = 0

DFS(1)
print(count)