1. 程式人生 > >leetcode 52 N皇後問題 II

leetcode 52 N皇後問題 II

n) vector spa 代碼 tco true .com abs leet

技術分享圖片

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

leetcode 52 N皇後問題 II