[leetcode]N-Queens

class Solution {
public:
bool col[100];
bool d1[100]; // i + j
bool d2[100]; // i - j + n
int cnt[100];

void nque(vector<vector<string> >& ans , int n , int dep) {
if(dep >= n) {
vector<string> xx(n);
for(int i = 0 ; i < n ; i++) {
string tmp = string(n , '.');
tmp[cnt[i]] = 'Q';
xx[i] = tmp;
}
ans.push_back(xx);
return ;
}

for(int i = 0 ; i < n ; i++) {
if(col[i] && d1[dep+i] && d2[dep-i+n]) {
cnt[dep] = i;
col[i] = false;
d1[dep+i] = false;
d2[dep-i+n] = false;
nque(ans , n , dep + 1);
col[i] = true;
d1[dep+i] = true;
d2[dep-i+n] = true;
}
}
}
vector<vector<string> > solveNQueens(int n) {
memset(col , true , sizeof(col));
memset(d1 , true , sizeof(d1));
memset(d2 , true , sizeof(d2));
vector<vector<string> > ans;
nque(ans , n , 0);
return ans;
}
};