【回溯】【DFS】51.N皇后

题目

java 复制代码
class Solution {
    public List<List<String>> solveNQueens(int n) {
        List<List<String>> res = new ArrayList<>();
        char[][] tmp = new char[n][n];
        for (int i = 0; i < n; ++i) {
            Arrays.fill(tmp[i], '.');
        }
        dfs(tmp, 0, res);

        return res;
    }

    public void dfs(char[][] tmp, int layer, List<List<String>> res) {
        int n = tmp.length;
        if (layer == n) {
            List<String> solu = new ArrayList<>();
            for (int k = 0; k < n; ++k) {
                solu.add(String.valueOf(tmp[k]));
            }
            res.add(solu);
            return;
        }
        for (int k = 0; k < n; ++k) {
            if (isValid(tmp, layer, k)) {
                tmp[layer][k] = 'Q';
                dfs(tmp, layer + 1, res);
                tmp[layer][k] = '.';
            }
        }
    }

    boolean isValid(char[][] tmp, int x, int y) {
        int n = tmp.length;
        for (int i = 0; i < n; ++i) {
            if (tmp[i][y] == 'Q' && i < x) {
                return false;
            }
        }
        int i = 1;
        while (x - i >= 0 && y - i >= 0) { 
            if (tmp[x - i][y - i] == 'Q') {
                return false;
            }
            ++i;
        }
        i = 1;
        while (x - i >= 0 && y + i < n) {
            if (tmp[x - i][y + i] == 'Q') {
                return false;
            }
            ++i;
        }
        
        return true;
    }
}
相关推荐
一只鱼^_1 天前
基础算法合集-图论
数据结构·算法·深度优先·图论·广度优先·宽度优先·图搜索算法
稳兽龙11 天前
P3258 [JLOI2014] 松鼠的新家
数据结构·c++·算法·深度优先·lca
xwztdas12 天前
洛谷P3953 [NOIP 2017 提高组] 逛公园
算法·深度优先·动态规划
小猫咪怎么会有坏心思呢13 天前
华为OD机考-小明减肥-DFS(JAVA 2025B卷)
java·华为od·深度优先
可可格子衫16 天前
129. 求根节点到叶节点数字之和 --- DFS +回溯(js)
javascript·算法·深度优先
小猫咪怎么会有坏心思呢16 天前
华为OD机试-返回矩阵中非1的元素、个数/数值同化-BFS(JAVA 2025B卷)
java·华为od·深度优先
程序员Xu20 天前
【大厂机试题解法笔记】恢复数字序列
笔记·算法·深度优先
别来无恙14921 天前
岛屿周长问题的三种解法:直接计数法、数学计算法与深度优先搜索
java·c++·算法·深度优先·dfs
青山是哪个青山24 天前
递归,回溯,DFS,Floodfill,记忆化搜索
算法·深度优先
jndingxin25 天前
c++ 面试题(1)-----深度优先搜索(DFS)实现
c++·算法·深度优先