【LeetCode笔记】51. N 皇后（DFS、经典题）

题目描述

• 经典题了…但是大一敲过后就再没碰过了。结果面试还是会问= =，今天补一下～
  
  

思路 && 代码

• 还是挺清晰的，要点如下：
  1. 使用 DFS，找到一个了再继续找
  2. 对每一行，逐列进行可行点选择
  3. 放置点可行判断：【对低行进行】列、45度、135度判断
  4. 答案格式转换，char[][] 变成 List

class Solution {
    List<List<String>> ans = new ArrayList<>(); // 全局变量，记录答案
    char[][] graph;

    public List<List<String>> solveNQueens(int n) {
        // 1、地图初始化
        graph = new char[n][n];
        for(int i = 0; i < n; i++) {
            for(int j = 0; j < n; j++) {
                graph[i][j] = '.';
            }
        }

        // 2、DFS 获取答案
        dfs(0);
        return ans;
    }

    void dfs(int row) {
        // 1、递归结束，加入答案
        if(row == graph.length) {
            ans.add(array2List());
        }

        // 2、对当前行，逐列进行判断、DFS
        for(int col = 0; col < graph[0].length; col++) {
            if(judge(row, col)) {
                graph[row][col] = 'Q';
                dfs(row + 1);
                graph[row][col] = '.';
            }
        }
    }

    // 判断可行：行不用，因为一行只放一个。高行不用，因为还没放棋子
    boolean judge(int row, int col) {
        // 1、列
        for(int i = row - 1; i >= 0; i--) {
            if(graph[i][col] == 'Q') {
                return false;
            }
        }
        // 2、45度
        for(int i = row - 1, j = col - 1; i >= 0 && j >= 0; i--, j--) {
            if(graph[i][j] == 'Q') {
                return false;
            }
        }
        // 3、135度
        for(int i = row - 1, j = col + 1; i >= 0 && j < graph[0].length; i--, j++) {
            if(graph[i][j] == 'Q') {
                return false;
            }
        }
        return true;
    }

    // 转换答案格式。
    List<String> array2List() {
        List<String> list = new ArrayList<>();
        for(int i = 0; i < graph.length; i++) {
            list.add(String.valueOf(graph[i]));
        }
        return list;
    }
}