N皇后问题

8/N皇后问题

说明

1、这里使用回溯法求解求解8/n皇后问题，即从第1行第一列放置第1个皇后一直到第n行放置第n个皇后，如果在第k（k<=n）行所有位置都不能放置一个新的皇后则返回第k-1行重新放置第k-1个皇后,直到所有皇后放置完。

2、任意两个皇后不能在同一行，a[n] = m 表示为第n个皇后放置在第n行第m列。

非递归版本

public class Main {
  public static void main(String[] args) {
    nQueen(8);                              //输入皇后数，这里为8皇后
  }

  public static void nQueen(int n) {
    int[] a = new int[n];                   // a[n] = m 表示为第n个皇后放置在第n行第m列
    int sum = 0;                            //记录所找到的个数

    int i = 0 ;                             //判断正在放置第几个皇后从第0个开始到n-1个
    while(i>=0) {   
        for(a[i]++ ; a[i]<=n ;a[i]++ ) {    //寻找一个位置放置皇后（从第1列到第n列）
            if(Check(a, i))
                break;
        }

        if(a[i]<=n) {                   //找到一个位置
            if(i == n-1) sum++;             //对找到的位置进行判断是否为最后一个数
            else i++;                       //如果不是最后一个数，继续找
        }else {                         //未能找到一个位置
            a[i] = 0;                       //考虑到退回故清理a[i]
            i--;
        }
    }
    System.out.println(sum);            //输出结果
  }

    public static boolean Check(int a[], int n) {   //检测第n个皇后放置位置是否与之前皇后冲突
        for (int i = 0; i < n; i++) {
            if (Math.abs(a[i] - a[n]) == Math.abs(i - n) || a[i] == a[n])
                return false;
        }
        return true;
    }
}

递归版本

public class Main{
    public static int n = 8;            // 皇后数目
    public static int[] r = new int[n]; // r[n] = m 表示为第n个皇后放置在第n行第m列
    public static int sum = 0;          // 结果数

    public static void main(String[] args) {
        nQueen(0);                      // 从第0个皇后开始放置
        System.out.println(sum);
    }

    public static void nQueen(int k) {  // 递归处理
        if (k == n) {                   // 已找到结果，记录，不再递归
            sum++;
            return;
        }

        for (r[k] = 0; r[k] < n; r[k]++)    // 从第0列第7列寻找到一个可以放置的位置
            if (Check(r, k)) nQueen(k + 1); //如果该位置可以放置则递归寻找下一个
    }

    public static boolean Check(int a[], int n) { // 检测第n个皇后放置位置是否与之前皇后冲突
        for (int i = 0; i < n; i++) {
            if (Math.abs(a[i] - a[n]) == Math.abs(i - n) || a[i] == a[n])
                return false;
        }
        return true;
    }
}