【Leetcode】51. N-Queens

题目地址：

https://leetcode.com/problems/n-queens/

经典的$n$皇后问题。求具体方案。

DFS回溯+剪枝。代码如下：

class Solution {
 public:
  vector<vector<string>> solveNQueens(int n) {
    vector<vector<string>> res;
    if (!n) return res;
    vector<int> pos(n);
    dfs(0, pos, n, res);
    return res;
  }

  void dfs(int u, auto& pos, int n, auto& res) {
    if (u == n) {
      vector<string> v(n, string(n, '.'));
      for (int i = 0; i < n; i++) v[i][pos[i]] = 'Q';
      res.push_back(v);
      return;
    }
    static vector<bool> col(n), diag(2 * n - 1), udiag(2 * n - 1);
    for (int i = 0; i < n; i++) {
      if (col[i] || diag[u - i + n - 1] || udiag[u + i]) continue;
      pos[u] = i;
      col[i] = diag[u - i + n - 1] = udiag[u + i] = true;
      dfs(u + 1, pos, n, res);
      col[i] = diag[u - i + n - 1] = udiag[u + i] = false;
    }
  }
};

时间复杂度$O(n^{2}n!)$，类似于先生成全排列，再把答案加到res里去（需要$O(n^2)$，因为棋盘有两维）。空间复杂度$O(n)$。

因为有剪枝，所以时间复杂度不太容易估计出来。这里取的是上界，实际要比它小一些。