leetcode 52. N-Queens II(N皇后II)

leetcode 52. N-Queens II(N皇后II)_第1张图片

nxn的棋盘上要放n个皇后,皇后不能在同一行,同一列,两条对角线上。
因为皇后能横着走,竖着走,还能沿两个方向的对角线斜着走。

思路:
回溯。
保证不冲突即可,用boolean数组代表列,对角线,反对角线上有无冲突。
对角线diag的size为什么是2n-1, 因为row是0~n-1, col是0~n-1,
用row+col代表(row, col), 所以是0~2n-2,size则是2n-1。
同理antiDiag。

class Solution {
    boolean[] col;
    boolean[] diag;
    boolean[] antiDiag;
    
    public int totalNQueens(int n) {
        col = new boolean[n];
        diag = new boolean[2*n - 1];
        antiDiag = new boolean[2*n - 1];
	    return solve(0);
    }
    
    int solve(int row) {
	    int n = col.length;
        int count = 0;
        if(row == n) return 1;
        for(int column = 0; column < n; column++)           
            if(!col[column] && !diag[row + column] && !antiDiag[row - column + n - 1]){ 
                col[column] = diag[row + column] = antiDiag[row - column + n - 1] = true;
                count += solve(row + 1); 
                col[column] = diag[row + column] = antiDiag[row - column + n - 1] = false; 
            }                                
        return count;
    }
}

你可能感兴趣的:(leetcode,leetcode,算法)