LeetCode 51 N皇后 回溯

// 最近在面试，面试是在牛客网上在线编程，绝望的是：不支持debug...
// 对于习惯用ide debug的我，又一个耻辱柱，加油...
// 递归回溯，使用一个一位数组记录每行皇后放的位置，当前行放皇后时，
// 判断同一列是否有，以及两条斜对角线是否发生碰撞，
// 数组的下标相当于横坐标，数组里存的值相当于纵坐标
// 初始从1开始调用，如果ok，进入下一行，直到进入第n + 1行
// 细节还是挺多的，哎。。。当时面的时候只有大致框架，挺多细节性的小错误
// 加油,尽量少些bug啊！！！

public class Solution {
    int[] vis;

    int cnt = 0;
    int col = 0;

    public boolean ok(int pos, int x) { // 判断位置
        for (int i = 1; i < x; i++) {
            if (vis[i] == pos || Math.abs(vis[i] - pos) == Math.abs(i - x))
                return false;
        }
        return true;
    }

    String getString(int pos) {
        StringBuilder sb = new StringBuilder("");
        for (int i = 1; i < pos; i++)
            sb.append('.');
        sb.append('Q');
        for (int i = pos + 1; i <= col; i++)
            sb.append('.');
        return sb.toString();
    }

    public void dfs(int x) {
        if (x == col + 1) {
            cnt++;
            res = new ArrayList<>();
            for (int i = 1; i <= col; i++) {
                res.add(getString(vis[i]));
            }
            ans.add(res);
            return ;
        }


        for (int i = 1; i <= col; i++) {
            if (ok(i, x)) {
                vis[x] = i;
                dfs(x + 1);
                vis[x] = 0;
            }
        }
    }

    public int queens (int n) {
        vis = new int[n + 1];
        cnt = 0;
        col = n;
        dfs(1);
        return cnt;
    }

    public List<List<String>> solveNQueens(int n) {
        ans = new ArrayList<>();
        queens(n);
        return ans;
    }

    List<String> res;
    List<List<String>> ans;
}