leetcode52——N-Queens II

题目大意：输出N皇后问题有多少个可行解

分析：dfs回溯。

代码：

class Solution {
public:
int ans = 0;
int totalNQueens(int n) {
vector<int> cols(n);
dfs(0, n, cols);
return ans;
}
void dfs(int level, int n, vector<int>& cols) {
if (level == n) {
ans++;
return;
}
for (cols[level] = 0;cols[level] < n;cols[level]++) {
if (isOk(level, cols)) {
dfs(level + 1, n, cols);
}
}
}
bool isOk(int level, vector<int>& cols) {
for (int i = 0;i < level;i++) {
if (cols[i] == cols[level] || abs(cols[i] - cols[level]) == abs(i - level))
return false;
}
return true;
}
};