题目
题目描述】
在国际象棋棋盘上放置八个皇后,要求每两个皇后之间不能直接吃掉对方。
【输入】
(无)
【输出】
按给定顺序和格式输出所有八皇后问题的解(见样例)。
【输入样例】
(无)
【输出样例】
No. 1
1 0 0 0 0 0 0 0
0 0 0 0 0 0 1 0
0 0 0 0 1 0 0 0
0 0 0 0 0 0 0 1
0 1 0 0 0 0 0 0
0 0 0 1 0 0 0 0
0 0 0 0 0 1 0 0
0 0 1 0 0 0 0 0
No. 2
1 0 0 0 0 0 0 0
0 0 0 0 0 0 1 0
0 0 0 1 0 0 0 0
0 0 0 0 0 1 0 0
0 0 0 0 0 0 0 1
0 1 0 0 0 0 0 0
0 0 0 0 1 0 0 0
0 0 1 0 0 0 0 0
...以下省略
代码
#include<iostream>
using namespace std;
const int N = 10;
int chess[N]; //记录棋盘情况
bool col[N], dg[2*N], udg[2*N]; //标记列和主次对角线是否被用过
int cnt = 0;
// 输出一个解
void print() {
cnt++;
cout << "No. " << cnt << endl; // 输出方案编号
for (int i = 1; i <= 8; i++)
{
for (int j = 1; j <= 8; j++)
{
if (chess[j] == i) //要反过来
cout << "1 ";
else
cout << "0 ";
}
cout << endl;
}
}
//检查点(x,y)能否放皇后
bool check(int x, int y) {
return (!col[y]) && (!dg[x + y]) && (!udg[x - y + 7]);
}
//往点(x,y)放皇后,flag=false则进行回溯
void put(int x, int y, bool flag) {
if (flag) chess[x] = y;
col[y] = flag;
dg[x + y] = flag;
udg[x - y + 7] = flag;
}
//尝试往第i行放置皇后
void dfs(int i)
{
for (int j = 1; j <= 8; j++)
{
if (check(i, j))
{
put(i, j, true); //放置皇后
if (i == 8) print(); //8个皇后放置好,输出
else dfs(i + 1); // 没放完,继续在下一行放皇后
put(i, j, false); //回溯
}
}
}
int main() {
dfs(1); //从第一行开始放皇后
return 0;
}