LeetCode--N-Queens II

dfs

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;
    }
};