用深搜解决n皇后问题

问题描述：
在一个n*n的棋盘上放置n个棋子，使得每行每列和对角线上都只有一个棋子，求有多少种摆放方法。
例如n=4；其中一种情况如下：

解题思路：
  这是典型的递归问题，用DFS可以解决，由于每一行只能放一个，所以不需要一个格子一个格子去试探，可以一行一行去试探；用一个一维数组来表示棋盘的情况，数组下标表示行，数组元素的值表示在对应的行的那一列放棋子。每一行一共有n种（n列）放法，每一种放法都去尝试，也就是这个DFS有n个支路。放棋子前进行判断是否满足条件即可。

  判断是否每行每列和对角线上都只有一个棋子，有一个简单的方法。由于我们是一行一行去试探的，所以每一行上肯定只有一个棋子。我们只需要去检测每一列和对角线上的棋子即可；对角线的判断方法：主对角线上的每一个点，行号减列号的结果是相同的；副对角线上的每一个点，行号加列号的结果是相同的。如下图所示。

代码如下：

import java.util.*;

public class n皇后问题 {
	public static void main(String[] args){
		Scanner scanner = new Scanner(System.in);
		int n = scanner.nextInt();
		int[] arr = new int[n];
		for(int i=0; i<arr.length; i++){
			//为了避免检查时出现错误，所以不初始化为0
			//因为数组元素值是用来表示放在第几列的
			//用int的最大值表示没有棋子
			arr[i] = Integer.MAX_VALUE;
		}
		dfs(arr, 0);
		System.out.println(count);
	}
	static int count = 0; //全局变量，用于统计摆法
	
	private static void dfs(int[] arr, int row) {
		if(row == arr.length){//找到一种摆法
			count++;
			return;
		}
		for(int i=0; i<arr.length; i++){//尝试在一行的n个格子摆放棋子
			if(check(arr, row, i)){//判断是否满足条件
				arr[row] = i;//表示第row行的棋子放在第i列
				dfs(arr, row+1);
				arr[row] = Integer.MAX_VALUE;//回溯
			}
		}
	}

	private static boolean check(int[] arr, int x, int y) {
		for(int i=0; i<arr.length; i++){
			if(arr[i] == y)//检查列
				return false;
			if(i-arr[i] == x-y)//检查主对角线
				return false;
			if(i+arr[i] == x+y)//检查副对角线
				return false;	
		}
		return true;
	}
}