【回溯】【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;
    }
}
相关推荐
guozhetao2 天前
【ST表、倍增】P7167 [eJOI 2020] Fountain (Day1)
java·c++·python·算法·leetcode·深度优先·图论
YuTaoShao2 天前
【LeetCode 热题 100】131. 分割回文串——回溯
java·算法·leetcode·深度优先
zc.ovo4 天前
图论水题日记
算法·深度优先·图论
深兰科技5 天前
绿地集团携手深兰科技推动AI医诊大模型快速落地
人工智能·科技·深度优先·新质生产力·深兰科技·绿地集团·ai医诊大模型
熬了夜的程序员6 天前
【华为机试】547. 省份数量
算法·华为·矩阵·golang·深度优先
YuTaoShao6 天前
【LeetCode 热题 100】78. 子集——(解法二)回溯+选哪个
java·算法·leetcode·深度优先
qqxhb7 天前
零基础数据结构与算法——第五章:高级算法-回溯算法&子集&全排列问题
算法·回溯算法·全排列·n皇后·子集
秋说7 天前
【PTA数据结构 | C语言版】列出连通集
c语言·数据结构·算法·深度优先·图论
wait a minutes8 天前
【c++】200*200 01灰度矩阵求所有的连通区域坐标集合
c++·矩阵·深度优先
熬了夜的程序员8 天前
【华为机试】240. 搜索二维矩阵 II
线性代数·算法·华为·面试·矩阵·golang·深度优先