回溯——八皇后问题

79 阅读1分钟

代码

public class EightQueen{
    //定义一个解的编号
    private static int count = 0;
    //定义棋盘的尺寸
    private static final int SIZE=8;
    public static void main(String[] args){
        int[][] arr=new int[SIZE][SIZE];
        eightQueen(0,arr);
    }
    public static void eightQueen(int row,int[][] arr){
        if(row==SIZE){
            count++;
            System.out.print("这是第"+count+"中解法:");
            print(arr);
        }else{
            int[][] newArr=copyOf(arr);
            for(int col = 0;col<SIZE;col++){
                if(noDangerous(row,col,newArr)){
                    //先清空之前同级别的可能
                    for(int c=0;c<SIZE;C++){
                        newArr[row][c]=0;
                    }
                    newArr[row][col]=1;
                    eightQueen(row+1,newArr);
                }
                
            }
        }
    } 
    
    //打印
    public static void print(int[][] arr){
        for(int i=0;i<SIZE;i++){
            for(int j=0;j<SIZE;j++){
                System.out.print(arr[i][j]+" ");
            }
             System.out.println();
        }
    }
    
    //复制
     public static int[][] copyOf(int[][] arr){
         int[][] newArr=new int[SIZE][SIZE];
        for(int i=0;i<SIZE;i++){
            for(int j=0;j<SIZE;j++){
                newArr[i][j]=arr[i][j];
            }
        }
        return newArr;
    }
    
    //判断是否安全
    public static boolean noDangerous(int row,int col,int[][] arr){
        //正上方
        for(int r=row-1;r>=0;r--){
            if(arr[r][col]==1){
                return false;
            }
        }
        //左上方
        for(int r=row-1,c=col-1;r>=0&&c>=0;r--,c--){
            if(arr[r][c]==1){
                return false;
            }
        }
        
        //右上方
        for(int r=row-1,c=col+1;r>=0&&c<SIZE;r--,c++){
            if(arr[r][c]==1){
                return false;
            }
        }
        return ture;
    }
}

分解是逐行向下进行分解,在循环里进行回溯。