Java与N皇后(leetcode第51)

leetcode第51题,N皇后。解决的核心思路是递归+回溯,与解数独那个思路类似。

class Solution {
    

    public List> solveNQueens(int n) {
        List> list = new ArrayList<>();
        char [][]cs = new char[n][n];
        for(int i = 0;i < n;i++){
            for(int j = 0;j < n;j++){
                cs[i][j] = '.';
            }
        }
        putQueen(list, 0, n, cs);
        return list;
    }

	// 尝试放置:递归+回溯
    private void putQueen(List> list, int row, int n,char [][]cs) {


        if (row == n) {
            List l = new ArrayList();
            for(int i = 0;i < n;i++){
                l.add(new String(cs[i]));
            }
            list.add(l);
            return;
        }

        for (int i = 0; i < n; i++) {
            for(int j = 0;j < n;j++){
                cs[row][j] = '.';
            }
            cs[row][i] = 'Q';
            if (isSafety(cs, row, i, n)) {
                putQueen(list, row + 1, n, cs);
            }
        }

    }

	// 检测放置的皇后是否有问题 
    private boolean isSafety(char[][]cs, int row, int col, int n) {
        int step = 1;
        while (row - step >= 0) {
            if (cs[row-step][col] == 'Q') {
                return false;
            }
            if (col - step >= 0 && cs[row-step][col-step] == 'Q') {
                return false;
            }
            if (col + step < n && cs[row-step][col+step] == 'Q') {
                return false;
            }
            step ++;
        }
        return true;
    }
    
}

你可能感兴趣的:(leetcode,leetcode,N皇后,八皇后,Java,回溯)