51的简化版,省去根据排列话棋盘的工作,直接计数,代码:
class Solution { public: int totalNQueens(int n) { int res=0; vector<int> pos(n,-1); dfs(n,0,pos,res); return res; } void dfs(int n,int row,vector<int>& pos,int &res){ if(row==n){ res++;return; } for(int col=0;col<n;col++){ if(isValid(row,col,pos)){ pos[row]=col; dfs(n,row+1,pos,res); pos[row]=-1; } } } bool isValid(int row,int col,vector<int>&pos){ for(int i=0;i<row;i++){ if(col==pos[i] || abs(col-pos[i])==abs(row-i)) return false; } return true; } };