P1219 [USACO1.5] 八皇后 Checker Challenge（回溯法的经典使用）

题目描述

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

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

行号 1 2 3 4 5 6

列号 2 4 6 1 3 5

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

输入格式

一行一个正整数 n，表示棋盘是 n×n 大小的。

输出格式

前三行为前三个解，每个解的两个数字之间用一个空格隔开。第四行只有一个数字，表示解的总数。

输入输出样例

输入 #1复制

6

输出 #1复制

2 4 6 1 3 5
3 6 2 5 1 4
4 1 5 2 6 3
4

说明/提示

【数据范围】
对于 100%100% 的数据，6≤n≤13。

题目翻译来自NOCOW。

USACO Training Section 1.5

本题是回溯法的经典使用例题，思路如下

检索整个棋盘，以列为宽度，以行为深度，一行一行地摆放皇后。

摆放好就row+1，进入下一层搜索

摆放前检验格子的合法性，这里每一列只能有一个皇后，而且之前摆放过皇后的格子的两个对角线也不能有皇后。

如果发现摆放完后，下一行一个皇后也放不下了（全都不合法）那么回到上一层的dfs，并把之前摆的皇后收起来，循环col++，继续往下一列搜索

这里存放数据有一个小细节，笔者一开始想只使用二维数组，用第二维存答案，每找到一种方案，就新开一个棋盘，后来发现这是非常错误的，因为新棋盘有 一个皇后都没有，但是我们的检索范围已经 到了棋盘底部了，就会 产生很多错解，正确的操作应该是单独建一个二维数组存  。不破坏原有的棋盘。希望大家能注意一下

还有就是数组初始化的时候不要带n，不然初始化不到，n没赋值的时候为0,

package dfs;


import java.util.Iterator;

import java.util.Scanner;



public class 八皇后速写 {
	static int n;
	static int [][]res=new int [100000][15];
	static int ans=0;
	static int []a=new int [15];//不能放n 来不及初始化，只会开0个格子
	
	
 public static void main(String[] args) {
	Scanner sc=new Scanner(System.in);
	n=sc.nextInt();
	dfs(0);
	
	//
	for(int i=0;i<3;i++)
	{
		for(int j=0;j<n;j++)
		System.out.print(res[i][j]+1+" ");
		System.out.println();
	}
	System.out.println(ans);
}//row 摆到第几行了
 //ans 摆好几个棋盘了
 static void dfs(int row)
 {
	 if(row==n) {
	 for(int j=0;j<n;j++)
	 {
		 res[ans][j]=a[j];//把答案拷贝过去//放弃使用二维数组
	 }
	 ans++;
	 return;}//找到正确，怎么更新下一盘呢
	 for(int col=0;col<n;col++)
	 {
		 if(isvaild(row,col))
		 {
			 a[row]=col;
			 dfs(row+1);
			// a[ans][row]=0;
		 }
	 }
 }//合法性检查 ，为什么有0031
 //用二维数组更新棋盘的问题，所有的皇后全丢失，所以应该是记录修改而不是全部清空
 static boolean isvaild(int row,int col)
 {
	 for(int i=0;i<row;i++)
	 {
	 if(a[i]==col) return false;//当前列有皇后
	 if(Math.abs(col-a[i])==row-i) return false;
	 }//利用斜率检查对角线 
	 return true;
 }
}