棋盘挑战（八皇后类似问题）题解 【DFS】

棋盘挑战

给定一个 N×N 的棋盘，请你在上面放置 N 个棋子，要求满足：

每行每列都恰好有一个棋子
每条对角线上都最多只能有一个棋子

    1   2   3   4   5   6
  -------------------------
1 |   | O |   |   |   |   |
  -------------------------
2 |   |   |   | O |   |   |
  -------------------------
3 |   |   |   |   |   | O |
  -------------------------
4 | O |   |   |   |   |   |
  -------------------------
5 |   |   | O |   |   |   |
  -------------------------
6 |   |   |   |   | O |   |
  -------------------------

上图给出了当 N=6 时的一种解决方案，该方案可用序列 2 4 6 1 3 5 来描述，该序列按顺序给出了从第一行到第六行，每一行摆放的棋子所在的列的位置。

请你编写一个程序，给定一个 N×N 的棋盘以及 N 个棋子，请你找出所有满足上述条件的棋子放置方案。

输入格式
共一行，一个整数 N。

输出格式
共四行，前三行每行输出一个整数序列，用来描述一种可行放置方案，序列中的第 i 个数表示第 i 行的棋子应该摆放的列的位置。

这三行描述的方案应该是整数序列字典序排在第一、第二、第三的方案。

第四行输出一个整数，表示可行放置方案的总数。

数据范围
6≤N≤13

输入样例：
6
输出样例：
2 4 6 1 3 5
3 6 2 5 1 4
4 1 5 2 6 3
4

分析

我们可以发现这一题其实就是N皇后问题的一个体现，遇到这样的问题我们就只能去深搜+剪枝来获得我们想要的所有答案了①我们可以以层数为形参进行深搜②分别拿三个数组来分别存储我们现在放的这个位置的同一列，还有两个对角线是否存在有棋子，下面来看一张图👇

来看代码吧，看完应该就明白上面的意思了

---

代码

#include<iostream>
#include<algorithm>
#include<string>
using namespace std;
const int N = 15;
bool col[N];//用于判断行是否给填了
bool d[N * 2];//判断斜率为正数的对角线
bool ud[N * 2];//判断斜率为负数的对角线
int res[N];
int ans;
int n;
void dfs(int x) {
	//cout << x << endl;
	//说明已经填完所有数了
	if (x > n) {
		ans++;
		if (ans <= 3) {
			for (int i = 1; i <= n; i++) {
				cout << res[i] << " ";
			}
			cout << endl;
		}
		return;
	}
	for (int y = 1; y <= n; y++) {
		if (!col[y] && !d[y - x + n] && !ud[x + y]) {
			col[y] = d[y - x  + n] = ud[x + y] = true;
			res[x] = y;
			dfs(x + 1);
			col[y] = d[y - x + n] = ud[x + y] = false;
			res[x] = 0;
		}
	}

}

int main() {
	cin >> n;
	dfs(1);//从第一层开始

	cout << ans << endl;
	return 0;
}

---

运行结果