LeetCode--N-Queens II

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