DFS_5:八皇后

DFS_5:八皇后

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

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

行号 1 2 3 4 5 6

列号2 4 6 1 3 5

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

输入格式
一行一个正整数 nn，表示棋盘是 n×n 大小的。

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

输入输出样例

6

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

问题分析

八皇后问题最难的点个人感觉就是找对角线的条件和记录路径如何表示：
u[x-i+n] == 0
v[x+i] == 0
a[x] = i ;
如何找出两条对角线的条件：
假设一个草图：
在两条对角线上随意找两三个点，你就会发现行-列是定值或者行+列是定值，那么。比如x=1，y=3和x=3，y=1处于一条对角线，此时的x+y为定值4则表示了两点所在的对角线，其他同样，那么x+y就可以当做一条对角线来使用。
特别是对于u[x-i+n] == 0为什么要加上n，因为x-i可能会越界，那么加上它的一个模，防止越界。

代码：

import java.util.Scanner;

public class DFS_4 {
    static  int ans = 0;
    static int  N = 20;
    static int n;
    static int[] column = new int[N];
    static int[] u = new int[N*2];
    static int[] v = new int[N*2];
    static int[] a = new int[N];
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        n = scanner.nextInt();
        //和全排列一样：这里传入的 1 标识第几层
        dfs(1);
        System.out.println(ans);

    }

    public static void dfs(int x) {
        //边界条件：大于n层结束
        if(x > n){
            ans++;
            print();
            return;
        }
        //每一层都有n种选择
        for (int i = 1; i <= n; i++) {
            //可达的条件
            if(column[i] == 0 && u[x-i+n] == 0 && v[x+i] == 0){
                column[i] = 1;
                u[x-i+n] = 1;
                v[x+i] = 1;
                //a[x]=i是用来记录路径的
                //表示的意思是：在X层所选择的列是第i列
                a[x] = i ;
                dfs(x+1);
                //回溯
                column[i] = 0;
                u[x-i+n] = 0;
                v[x+i] = 0;
            }
        }
    }

    public static void print() {
        if(ans<=3){
            for (int i = 1; i <= n  ; i++) {
                System.out.print(a[i]+" ");
            }
            System.out.println();
        }
    }
}