[Leetcode]N-Queens II

最新推荐文章于 2022-06-06 10:35:33 发布

dyllanzhou

最新推荐文章于 2022-06-06 10:35:33 发布

阅读量300

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/dyllanzhou/article/details/50318203

DFS 同时被 3 个专栏收录

28 篇文章

订阅专栏

Backtracking

22 篇文章

订阅专栏

Depth-first Search

17 篇文章

订阅专栏

本文介绍了一种使用深度优先搜索(DFS)算法解决N皇后问题的方法，并实现了计算所有可能解决方案数量的功能。通过递归地寻找每个皇后放置的位置，确保它们不在同一行、列或对角线上。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Follow up for N-Queens problem.

Now, instead outputting board configurations, return the total number of distinct solutions.

class Solution {
public:
    /*algorithm: DFS
    */
    vector<int>getNextCandiates(vector<int>&solution,int n)  
    {  
        vector<int>result;  
        for(int i = 0; i < n;i++){  
           bool selected = true;  
           for(int k = 0;k < solution.size();k++){  
               if(i == solution[k] ||  //same column or same k=1/-1 (k=(y2-y1)/(x2-x1))  
                  abs(i-solution[k])==abs((int)solution.size()-k)){   
                      selected = false;  
                      break;  
                  }  
           }  
           if(selected)result.push_back(i);  
        }  
        return result;  
    }  
     
    void dfs(int n,int start,int &sum,vector<int>&solution){  
        if(start == n){ //find one solution,reach last line  
            sum++; 
            return;  
        }  
        vector<int>candiates = getNextCandiates(solution,n);  
        for(int k = 0;k < candiates.size();k++){  
            solution.push_back(candiates[k]);  
            dfs(n,start+1,sum,solution);  
            solution.pop_back();  
        }  
    }  
    int totalNQueens(int n) {
        vector<int>solution;  
        int sum = 0;  
        dfs(n,0,sum,solution);  
        return sum;        
    }
};