Leetcode 51. N-Queens

q_pos 来记录 每个Queen 在每列的位置

public List<List<String>> solveNQueens(int n) {  
        List<List<String>> res = new ArrayList<List<String>>();          
        placeQueen(new ArrayList<>() , n, res, 0);
        return res;  
    }  
    
    private void placeQueen(List<Integer> q_pos, int n, List<List<String>> res, int row) {
        if (q_pos.size() == n) {
            List<String> board = new ArrayList<>();
            for (Integer pos : q_pos) {
                char[] line = new char[n];
                Arrays.fill(line, '.');
                line[pos] = 'Q';
                board.add(String.valueOf(line));
            }
            res.add(board);
            return;
        }
        
        for (int col = 0; col < n; col++) {
            if (!isValid(q_pos, row, col)) continue;   ／／ 如果发生了冲突的话就跳过这种情况
            q_pos.add(col);
            placeQueen(q_pos, n, res, row + 1);  ／／ row + 1 因为 如果一行有了 一个Queen的话就放不了 另外一个 Queen
            q_pos.remove(q_pos.size() - 1); ／／ 使用回溯法的话，要去掉最后一个
        }
    }
    
      
    private boolean isValid(List<Integer> q_pos, int x, int y) {  
        for (int i = 0; i < q_pos.size(); i++) {
            if (y == q_pos.get(i)) return false;     ／／ 此列有了 Queen，就返回 false
            if (x - y == i - q_pos.get(i)) return false; ／／ 检查对角线
            if (x + y == i + q_pos.get(i)) return false;
        }
        return true;
    }