题解 | #N皇后问题#

N皇后问题

https://www.nowcoder.com/practice/c76408782512486d91eea181107293b6

#include <vector>
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param n int整型 the n
     * @return int整型
     */
    int res = 0;
    bool check(vector<int> vec, int x, int y){
        // 如果在同一列,即vec中存在相同的数是不行的,另外要确保是在不同的斜线上
        for(int i = 0; i < vec.size() - 1; i++){
            if(y == vec[i] || y - x == vec[i] - i - 1 || x + y == vec[i] + i + 1) return false;
        }
        return true;
    }
    void cal(vector<int>& vec, int n){
        // 可以不在这里做检查,但是当vec的size为4的时候就可以res++
        if(vec.size() == n) {
            res++;
            return;
        }

        for(int i = 0; i < n; i++){
            vec.push_back(i+1);
            if(check(vec, vec.size(), i+1)) cal(vec, n);
            vec.pop_back();
        }
    }
    int Nqueen(int n) {
        // write code here
        // 主要思路---采用回溯算法,使用一个数组记录每个节点所在的位置
        // ---- 在每次递归回溯前都要进行一次判断,可以通过节点所在的情况进行判断
        vector<int> vec;
        cal(vec, n);
        return res;
    }
};

清晰明了的递归思路

全部评论

相关推荐

点赞 评论 收藏
分享
孤寡孤寡的牛牛很热情:为什么我2本9硕投了很多,都是简历或者挂,难道那个恶心人的测评真的得认真做吗
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务