洛谷P1219 [USACO1.5]八皇后 Checker Challenge

拖了很久的八皇后问题，这还是简化版的题目，简单记录一下,虽然能写出来代码，但八皇后的问题解决的还是不够清晰，没有上一道三连击题目清晰，下面贴出题目：

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

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

行号 1 2 3 4 5 6
列号 2 4 6 1 3 5

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

解法思路：
1.首先考虑输入和输出，需要输入棋盘的大小，在main函数里面先进行
cin >> a;
确认好棋盘的大小。

2.其次考虑用什么方法来实现这种输出，典型的dfs问题，假设是6*6的棋盘，首先是利用回溯法先输出6×6所有的序列，也就先输出全排列，然后进行条件判断，若条件满足，则输出满足条件的序列即可。

下面贴上代码

int used[13] = { 0 };
int result[13] = { 0 };
int a,sum;

bool judge()
{
	for (int i = 0; i < a; i++)
	{
		for (int j = i + 1; j < a; j++)
		{ 
			if(abs(result[i] - result[j]) == abs(i - j))
				return false;
		}
	}
	sum++;
	return true;
}

void dfs(int count)
{
	if (count == a)
	{
		if (judge() && sum < 4)
		{
			for (int j = 0; j < a; j++)
			{
				cout << result[j] << " ";
			}
			cout << endl;
		}
	}
	for (int i = 0; i < a; i++)
	{
		
		if (used[i] == 0)//配合上剪枝使用
		{
			int flag = true;
			for (int k = 0; k < count; k++)
			{
				if (abs(i + 1 - result[k]) == abs(k-count))flag = false;
			}
			if (flag == true)
			{
				used[i] = 1;
				result[count] = i + 1;
				dfs(count + 1);
				used[i] = 0;
			}
		}
	}
}
int main()
{
	cin >> a;
	dfs(0); 
	cout << sum;
	return 0;
}

有个需要注意的小细节，因为数组一开始就要定义好，所以直接设成最大的即可，这也是一种避免程序错误的方法，而该题目在判断两个皇后是否在同一斜线上是用，两个皇后的行减行，列减列来进行判断的。也就是

if(abs(result[i] - result[j]) == abs(i - j))

其中result是用来保存皇后的位置的。

for (int i = 0; i < a; i++)
	{
		
		if (used[i] == 0)//配合上剪枝使用
		{
			int flag = true;
			for (int k = 0; k < count; k++)
			{
				if (abs(i + 1 - result[k]) == abs(k-count))flag = false;
			}
			if (flag == true)
			{
				used[i] = 1;
				result[count] = i + 1;
				dfs(count + 1);
				used[i] = 0;
			}
		}
	}

下面的是我的写法，其中我的写法是没有剪枝的，所以我的复杂度很高，剪枝的方法还是不太熟悉，我需要特别的加强，看了半天代码也没看明白，还是多做点回溯题来慢慢熟悉吧

for (int i = 0; i < a; i++)
	{
		if (used[i] == 0)
		{
			used[i] = 1;
			result[count] = i + 1;
			dfs(count + 1);
			used[i] = 0;
		}
	}