N-Queens II

Follow up for N-Queens problem.

Now, instead outputting board configurations, return the total number of distinct solutions.

跟N-Queens 一个解发 最后return result.size()

public class Solution{

    public ArrayList<String[]> solveNQueens(int n) {  
        ArrayList<String[]> result = new ArrayList<String[]>();  
        int[] queenList = new int[n];  
        placeQueen(queenList, 0, n, result);  
        return result;  
    }  
       
    // 递归回溯8皇后，关键记录下到达了哪一行了  
    public void placeQueen(int[] queenList, int row, int n, ArrayList<String[]> result){  
        // Base Case, 已经完成任务了  
        if(row == n){  
            StringBuilder[] sol = new StringBuilder[n];  
               
            // 对数组内每一个对象都要new出其对象, 并将其全设为'.' 
            for(int i=0; i<n; i++){  
                sol[i] = new StringBuilder();  
                for(int j=0; j<n; j++){  
                    sol[i].append(".");  
                }  
            }  
            // 在相应的地方放置queen  
            for(int i=0; i<n; i++){  
                sol[i].setCharAt(queenList[i], 'Q');  
            }  
            String[] ss = new String[n];  
            for (int i=0; i<n; i++) {  
                ss[i] = sol[i].toString();  
            }  
            result.add(ss);  
            return;  
        }  
           
        // 开始这一行的查找  
        // 遍历第row行的所有列，测试哪一个位置是安全的,然后一行一行的放queen 
        for(int col=0; col<n; col++){  
            if(isSafe(queenList, row, col)){  
                queenList[row] = col;  
               //递归到下一行
                placeQueen(queenList, row+1, n, result);  
            }  
        }  
    }  
       
    // 判断是否坐标(row,col)的位置是安全的（检查行，列，正反对角线）  
    // queenList里面存放行，列坐标pair，即queenList[row] = col  
    public boolean isSafe(int[] queenList, int row, int col){
      //遍历当前row之前的所有row
        for(int preRow=0; preRow<row; preRow++){  
            // 得到上一行queen 放在哪一个 col里
            int preCol = queenList[preRow];  
            if(preRow == row){      // 理论上不必检查，因为preRow是总是小于row的  
                return false;  
            }  
            if(preCol == col){          // 检查是否在同一列  
                return false;  
            }  
            if(row-preRow == col-preCol){       // 反对角线  
                return false;  
            }  
            if(preRow+preCol == row+col){       // 正对角线  
                return false;  
            }  
        }  
        return true;  
    }      

}