蓝桥杯 n皇后问题 DFS Java

题目描述

在 N×N 的方格棋盘放置了 N 个皇后，使得它们不相互攻击（即任意 2 个皇后不允许处在同一排，同一列，也不允许处在与棋盘边框成 45 角的斜线上。你的任务是，对于给定的 N，求出有多少种合法的放置方法。

输入描述
输入中有一个正整数 N≤10，表示棋盘和皇后的数量

输出描述
为一个正整数，表示对应输入行的皇后的不同放置数量。

输入

5
 输出

10

运行限制

• 最大运行时间：1s
• 最大运行内存: 128M

找位置，需要回溯所以用DFS。先放第一个皇后如果与之前放置的皇后不在同一行同一列同一斜线上继续深入，不满足条件就回溯，直至放置的皇后数量大于N。

每当放置的皇后数量大于N时，记录符合的方案数。

import java.util.Scanner;

public class Main {
	static int N;//N行N列
	static int res;//方案数
	static int col[];//为每个皇后放置的列数，例如col[1]=1说明第一个皇后放在第1行的第1列
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		N = sc.nextInt();
		col = new int[N+1];
		dfs(1);//先放第一个皇后
		System.out.println(res);
	}
	private static void dfs(int k) {
		if(check(k)){//k为放置的皇后个数也是行数，当k >N时说明该方案符合要求，res自加1
			return;
		}
		for(int i =1;i<=N;i++){
			col[k] = i;//遍历列数，将k个皇后放入1到N列其中之一
			if(pd(k)){//判断是否与之前的皇后同行同列同斜线
				dfs(k+1);//满足条件，安排下一个皇后
			}else{
				continue;
			}
		}
	}
	private static boolean pd(int k) {
		for(int i = 1;i< k;i++ ){
			if(Math.abs(k - i) == Math.abs(col[k] - col[i])){
				return false;
			}else if(col[k] == col[i]){
				return false;
			}
		}
		return true;
	}
	private static boolean check(int k) {
		if(k > N){
			res ++;
			return true;
		}
		return false;
	}

}