N皇后和N皇后2

leetcode 51.N皇后

回溯法：
dfs:从第1行遍历到第n行。每一行中遍历所有可能放置的列，利用isUsable判断是否是可行的皇后位置：如果是当前位置放皇后，继续dfs，最后需要回溯，将当前位置的皇后去掉。当第n行遍历结束时，用track转换为题目的输出。
isUsable：判断当前位置能否放置皇后，需要判断当前列和左上，右上即可。左下和右下，因为没有放置，判断没有意义。
track：将二维int数组转换为题目的输出。

class Solution {
   List<List<String>> res=new ArrayList<>();
public List<List<String>> solveNQueens(int n) {
    //棋盘,默认为0表示空，1表示皇后
    int[][] board=new int[n][n];
    //row当前填写得的行号
    dfs(n,0,board);
    return res;
}

//深度优先遍历
private void dfs(int n, int row, int[][] board) {
    //0~n-1都填写完毕
    if (row==n){
        res.add(track(board,n));
        return;
    }
    for (int col = 0; col < n; col++) {
        if (isUsable(board,row,col)){
            board[row][col]=1;
            //填写下一行
            dfs(n,row+1,board);
            board[row][col]=0;//回溯
        }
    }
}

//board[row][col]是否可用,只需判断之前的是否满足就可以
private boolean isUsable(int[][] board, int row, int col) {
    //检查列上有无皇后
    for (int i = 0; i < row; i++) {
        if (board[i][col]==1)return false;
    }
    //判断当前坐标的右上角有没有皇后
        for (int i = row - 1, j = col + 1; i >= 0 && j < board.length; i--, j++) {
            if (board[i][j] ==1) {
                return false;
            }
        }
        //判断当前坐标的左上角有没有皇后
        for (int i = row - 1, j = col - 1; i >= 0 && j >= 0; i--, j--) {
            if (board[i][j] ==1) {
                return false;
            }
        }
    return true;
}

//将int类型棋盘转换成输出格式
private List<String> track(int[][] board, int n) {
    List<String> list=new ArrayList<>();
    for (int i = 0; i < n; i++) {
        StringBuilder temp=new StringBuilder();
        for (int j = 0; j < n; j++) {
            if (board[i][j]==0)temp.append('.');
            else temp.append('Q');
        }
        list.add(temp.toString());
    }
    return list;
}
}

leetcode 53.N皇后2
和1的不同之处在于，1需要输出完整的结果，2只需要输出个数。因此2对比1来说，没有输出转换函数，也不需要结果集，只需在第n行遍历结束时候，计数+1即可。

class Solution {
    int count=0;
    public int totalNQueens(int n) {
        int[][] board=new int[n][n];
        dfs(board,0,n);
        return count;
    }
     private void dfs(int[][] board,int row,int n){
        if(row==n){
        count++;
        return;
        }
        for(int col=0;col<n;col++){
            if(isUseable(board,row,col)){
            board[row][col]=1;
            dfs(board,row+1,n);
            board[row][col]=0;
            }
        }
    }
    private boolean isUseable(int[][] board,int row,int col){
        for(int i=0;i<row;++i){
            if(board[i][col]==1)
            return false;
        }
        for(int i=row-1,j=col-1;i>=0&&j>=0;--i,--j){
            if(board[i][j]==1)
            return false;
        }
        for(int i=row-1,j=col+1;i>=0&&j<board.length;--i,++j){
            if(board[i][j]==1)
            return false;
        }
        return true;
    }
}