力扣：51. N 皇后

最新推荐文章于 2025-07-22 18:03:35 发布

what is an object

最新推荐文章于 2025-07-22 18:03:35 发布

阅读量44

点赞数

CC 4.0 BY-SA版权

分类专栏：菜鸟刷题记录算法文章标签：大数据

本文链接：https://blog.youkuaiyun.com/newOneObject/article/details/125838950

算法同时被 2 个专栏收录

278 篇文章

订阅专栏

菜鸟刷题记录

204 篇文章

订阅专栏

本文介绍了一种使用Java实现的N皇后问题解决方案，通过递归和回溯法寻找合法布局，详细展示了如何初始化dp数组、处理约束条件并构建棋盘。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

class Solution {
    
    int[] dp;
    List<List<String>> list ;
    Set<Integer> x ;
    Set<Integer> y;
    Set<Integer> z;
    public List<List<String>> solveNQueens(int n) {
        dp = new int[n];
        // 初始化
        Arrays.fill(dp, -1);
        list = new ArrayList<>();
        x = new HashSet<>();
        y = new HashSet<>();
        z = new HashSet<>();
        calculation(n, 0, dp, x, y ,z);
        return list;
    }

    private void calculation(int n, int row, int[] dp, Set<Integer> x, Set<Integer> y, Set<Integer> z){
        if (n == row){
            List<String> board = buildBoard(dp, n);
            list.add(board);
            return;
        } else {
            for (int i = 0 ; i < n; i++){
                if (x.contains(i)){
                    continue;
                }

                int y1 = row - i;
                if (y.contains(y1)){
                    continue;
                }

                int z1 = row + i;
                if (z.contains(z1)){
                    continue;
                }
                dp[row] = i;
                x.add(i);
                y.add(y1);
                z.add(z1);
                calculation(n, row + 1, dp, x, y, z);
                dp[row] = -1;
                x.remove(i);
                y.remove(y1);
                z.remove(z1); 
            }
        }
    }

    private List<String> buildBoard(int[] dp, int n){
        List<String> res = new ArrayList<>();
        for (int i = 0; i < n; i++){
            char[] board = new char[n];
            Arrays.fill(board, '.');
            board[dp[i]] = 'Q';
            res.add(new String(board));
        }
        return res;
    }
}