leetcode_51. N 皇后

51. N 皇后

题目描述：按照国际象棋的规则，皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。

n 皇后问题 研究的是如何将 n 个皇后放置在 n×n 的棋盘上，并且使皇后彼此之间不能相互攻击。

给你一个整数 n ，返回所有不同的 n 皇后问题 的解决方案。

每一种解法包含一个不同的 n 皇后问题 的棋子放置方案，该方案中 'Q' 和 '.' 分别代表了皇后和空位。

输入：n = 4
输出：[[".Q..","...Q","Q...","..Q."],["..Q.","Q...","...Q",".Q.."]]
解释：如上图所示，4 皇后问题存在两个不同的解法。

示例 2：

输入：n = 1
输出：[["Q"]]

代码思路：

1. 使用回溯法：通过递归和回溯的方法来探索所有可能的皇后放置布局。回溯法通过尝试所有可能的位置来找到所有符合条件的解，并在不符合条件时回退到上一步进行其他尝试。

2. 递归方法 backTrack：

   1. 终止条件：当所有 N 行的皇后都成功放置时，构造当前棋盘的字符串表示并保存到结果列表中。
   2. 尝试放置皇后：在当前行的每一列上尝试放置皇后。

放置皇后时的注意事项：

• 合法性检查：在尝试放置皇后时，需要确保该位置不与已放置的皇后在同一列或对角线上。
• 状态更新：如果当前位置合法，将皇后放置在当前位置，并递归调用 backTrack 方法尝试在下一行放置皇后。
• 回溯：递归返回后，移除当前行的皇后，恢复到上一个状态，以尝试其他位置。

class Solution {
    List<List<String>> ans = new ArrayList<>();
    LinkedList<Integer> cur = new LinkedList<>();

    public List<List<String>> solveNQueens(int n) {
        backTrack(n, 0);
        return ans;
    }

    public void backTrack(int n, int index) {
        if (index >= n) {
            List<String> tem = new ArrayList<>();
            for (int i = 0; i < n; i++) {
                StringBuilder temp = new StringBuilder();
                for (int j = 0; j < n; j++) {
                    if (cur.get(i) == j) {
                        temp.append("Q");
                    } else {
                        temp.append(".");
                    }
                }
                tem.add(temp.toString());
            }
            ans.add(tem);
        }

        for (int i = 0; i < n; i++) {
            // 判断[index,i]能否放置棋子
            boolean flag = true;
            for (int j = 0; j < cur.size(); j++) {
                if (cur.get(j) == i) {
                    flag = false;
                }
                // [index,i] [j,cur.get(j)]
                if (Math.abs(j - index) == Math.abs(cur.get(j) - i)) {
                    flag = false;
                }
            }

            if (!flag) {
                continue;
            }
            cur.add(i);
            backTrack(n, index + 1);
            cur.removeLast();
        }
    }
}