Leetcode#51 N-Queens

原题地址

经典的8皇后问题变形版

当然也是用经典的8皇后问题解法的变形版咯（仿照"位运算求解8皇后问题"）

row：当前是第几行

col：皇后所在的列的情况

left：左上角禁位（因为我是从下向上迭代的）

right：右上角禁位（因为我是从下向上迭代的）

代码：

vector<vector<string> > result;
vector<string> buffer;

void solve(int row, vector<bool> &col, vector<bool> left, vector<bool> right) {
  if (row == 0) {
    result.push_back(buffer);
    return;
  }

  // 计算当前行可以放置皇后的位
  vector<bool> avail(col.size(), false);
  for (int i = 0; i < col.size(); i++)
    avail[i] = !(col[i] || left[i] || right[i]);

  // 计算上一行的左右禁位
  for (int i = 0; i < col.size() - 1; i++)
    left[i] = left[i + 1];
  left[col.size() - 1] = false;
  for (int i = col.size() - 1; i > 0; i--)
    right[i] = right[i - 1];
  right[0] = false;

  // 尝试、回溯
  for (int i = 0; i < col.size(); i++) {
    if (!avail[i])
      continue;
    col[i] = true;
    if (i > 0) left[i - 1] = true;
    if (i < col.size() - 1) right[i + 1] = true;
    buffer.push_back(string(i, '.') + "Q" + string(col.size() - i - 1, '.'));
    solve(row - 1, col, left, right);
    buffer.pop_back();
    if (i > 0) left[i - 1] = false;
    if (i < col.size() - 1) right[i + 1] = false;
    col[i] = false;
  }
}