码迷,mamicode.com
首页 > 其他好文 > 详细

leetcode 52 N皇后问题 II

时间:2019-04-12 23:05:37      阅读:185      评论:0      收藏:0      [点我收藏+]

标签:计数   cto   工作   void   png   直接   code   技术   solution   

技术图片

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

标签:计数   cto   工作   void   png   直接   code   技术   solution   

原文地址:https://www.cnblogs.com/joelwang/p/10699042.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!