N皇后算法
最近正在学习回溯法,遇到的第一个问题就是n皇后问题,问题如下:
要求在一个n×n的棋盘上放置n个皇后,使得任意两个皇后不在同一行或同一列或同一斜线上。
直接上代码:
#include<iostream>
#include<math.h>
using namespace std;
void NQueen(int, int, int*);
int main()
{
int x[4] = { -1, -1, -1, -1 }; //4 * 4的棋盘测试
NQueen(0, 4, x);
return 0;
}
bool Place(int k, int i, int* x) //用来判断皇后是否在同一列或者同一斜线,k表示当前行号,i记录当前的列号,x记录了当前棋盘信息
{
for (int j = 0; j < k; j++)
{
if ((x[j] == i) || abs(x[j] - i) == abs(j - k))
return false;
}
return true;
}
void NQueen(int k, int n, int* x)
{
for (int i = 0; i < n; i++) //遍历当前递归到行的所有列号
{
if(Place(k, i, x)) //判断当前位置是否可行
{
x[k] = i;
if (k == n - 1) //如果到了最后一行,说明是一个可行解,输出
{
for (int j = 0; j < n; j++)
{
cout << x[j] << " ";
}
cout << endl;
}
else //深度优先进入下一层递归
{
NQueen(k + 1, n, x);
}
}
}
return;
}


