N皇后问题

回溯法

 回溯法一般有两种实现方式，分别是递归回溯和迭代回溯。

递归：函数的定义中使用函数自身的方法。把问题看做是若干个相同解法的子问题

        （控制传入的参数来逐个解决子问题）

迭代：重发反馈过程的活动（循环）

NO.1

问题描述

　　给定一个n*n的棋盘，棋盘中有一些位置不能放皇后。现在要向棋盘中放入n个黑皇后和n个白皇后，使任意的两个黑皇后都不在同一行、同一列或同一条对角线上，任意的两个白皇后都不在同一行、同一列或同一条对角线上。问总共有多少种放法？n小于等于8。

输入格式

　　输入的第一行为一个整数n，表示棋盘的大小。
　　接下来n行，每行n个0或1的整数，如果一个整数为1，表示对应的位置可以放皇后，如果一个整数为0，表示对应的位置不可以放皇后。

输出格式

　　输出一个整数，表示总共有多少种放法。

样例输入

4
1 1 1 1
1 1 1 1
1 1 1 1
1 1 1 1

样例输出

2

样例输入

4
1 0 1 1
1 1 1 1
1 1 1 1
1 1 1 1

样例输出

0

先来看看单个的N皇后问题

import java.math.BigInteger;
import java.util.Scanner;

public class Main{
    static int n,count;
    static  int array[][];
    public static void main(String[] args) {
        Scanner sc=new Scanner(System.in);
         n=sc.nextInt();
        array=new int[n][n];
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                array[i][j]=sc.nextInt();
            }
        }

        put(0);
        System.out.println(count);
    }

    private static void put(int t){
        if(t==n){
            count++;
            return;
        }
        for(int i=0;i<n;i++){
            if(array[t][i]==0) continue;
            if(!check(t,i)) continue;
            array[t][i]=2;
            put(t+1);
            array[t][i]=1;
        }
    }

    private static boolean check(int t,int i){
        boolean b=false;
        for(int j=0;j<t;j++){
            if(array[j][i]==2) return b;
        }
        for(int j=1;t-j>=0&&i-j>=0;j++){
            if(array[t-j][i-j]==2) return b;
        }
        for(int j=1;t-j>=0&&i+j<n;j++){
            if(array[t-j][i+j]==2) return b;
        }
        b=true;
        return b;
    }

}

然后是2n皇后问题

import java.math.BigInteger;
import java.util.Scanner;

public class Main{
    static int n,count;
    static  int array[][];
    public static void main(String[] args) {
        Scanner sc=new Scanner(System.in);
         n=sc.nextInt();
        array=new int[n][n];
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                array[i][j]=sc.nextInt();
            }
        }

        put(0,2);
        System.out.println(count);
    }

    private static void put(int t,int h){
        if(t==n){
            if(h==2){
                put(0,3);
            }else count++;//注意这，别多加
            return;
        }
        for(int i=0;i<n;i++){
            if(array[t][i]!=1) continue;//注意这
            if(!check(t,i,h)) continue;

            array[t][i]=h;
            put(t+1,h);
            array[t][i]=1;
        }
    }

    private static boolean check(int t,int i,int h){
        boolean b=false;
        for(int j=0;j<t;j++){
            if(array[j][i]==h) return b;
        }
        for(int j=1;t-j>=0&&i-j>=0;j++){
            if(array[t-j][i-j]==h) return b;
        }
        for(int j=1;t-j>=0&&i+j<n;j++){
            if(array[t-j][i+j]==h) return b;
        }
        b=true;
        return b;
    }

}