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;
}
hmoban主题是根据ripro二开的主题,极致后台体验,无插件,集成会员系统
自学咖网 » N皇后算法