leetcode -：N-Queens

N皇后问题，回溯算法中的经典问题：

在一个N*N的棋盘中放置棋子，使其不在同一行同一列和对角线上。

暴力的做法，最普通的就是每个旗子的位置都遍历一次，即取与不取都选择，这样子复杂度是（2^(N*N)）

我的做法是借助全排列的，因为题目要求放置的棋子要求是不在同一行和同一列的，也就是我们使用数组arr[1~N]来表示第1~N行棋子是放置在第arr[1~N]，其中可以知道arr[]的范围也是1~N。这样子我们就可以把缩小到O(N!)

Java代码如下

 

public class Solution {
    static ArrayList<String[]> ans;
        static int[] number;

        public ArrayList<String[]> solveNQueens(int n) {
            ans = new ArrayList<>();
            number = new int[n + 1];
            for (int i = 0; i < number.length; ++i) {
                number[i] = i;
            }
            dfs(1);
            return ans;
        }

        void dfs(int cur) {
            if (cur == number.length) {
                char[] string = new char[number.length - 1];
                String[] stringes = new String[number.length - 1];
                Arrays.fill(string, '.');
                for (int i = 1; i < number.length; ++i) {
                    string[number[i] - 1] = 'Q';
                    stringes[i - 1] = new String(string);
                    string[number[i] - 1] = '.';
                }
                ans.add(stringes);
            } else {
                for (int i = cur; i < number.length; ++i) {
                    swap(number, cur, i);
                    if (isCan(cur)) {
                        dfs(cur + 1);
                    }
                    swap(number, cur, i);
                }
            }
        }

        void swap(int[] a, int i, int j) {
            int t = a[i];
            a[i] = a[j];
            a[j] = t;
        }

        boolean isCan(int cur) {
            for (int i = 1; i < cur; ++i) {
                if ((cur - i) == Math.abs(number[cur] - number[i])) {
                    return false;
                }
            }
            return true;
        }
}