Leetcode 51. N-Queens N皇后

题目：

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

上图为 8 皇后问题的一种解法。

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

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

示例:

输入: 4
输出: [
 [".Q..",  // 解法 1
  "...Q",
  "Q...",
  "..Q."],

 ["..Q.",  // 解法 2
  "Q...",
  "...Q",
  ".Q.."]
]
解释: 4 皇后问题存在两个不同的解法。

解题思路：

典型的回溯法问题，注意每次放置一个皇后的时候要判断是否合法。

代码实现：

class Solution {

    public List<List<String>> solveNQueens(int n) {

        // 初始化
        List<List<String>> result = new LinkedList<List<String>>();
        StringBuilder builder = new StringBuilder();
        for (int i = 0; i < n; i ++) {
            builder.append(".");
        }

        int[] line = new int[n];
        Arrays.fill(line, -1);
        int scale = 0, maxScale = n;

        // 回溯法
        // 终止条件
        while (scale >= 0) {

            // 输出条件：scale为maxScale
            if (scale == maxScale) {
                List<String> chess = new LinkedList<String>();
                for (int i = 0; i < scale; i ++) {
                    builder.setCharAt(line[i], 'Q');
                    chess.add(builder.toString());
                    builder.setCharAt(line[i], '.');
                }
                result.add(chess);
                scale --;
            }

            // 如何前进与后退
            line[scale] ++;
            // 如果合法，去下一行
            if (check(line, scale)) {
                scale ++;
            } else {
                // 不合法，再往下一列放，如果到底了，返回上一行
                if (line[scale] >= n) {
                    line[scale] = -1;
                    scale --;
                }
            }

        }

        return result;

    }

    // 检查放置位置是否合法
    private boolean check(int[] line, int scale) {
        int k = line[scale];
        if (k >= line.length) return false;
        if (scale == 0) return true;
        for (int i = 0; i < scale; i ++) {
            if (line[i] == k || Math.abs(line[i] - k) == Math.abs(i - scale)) return false;
        }
        return true;
    }

}