Leetcode 52: N皇后II

本文介绍了一种解决N皇后问题的方法,通过递归回溯算法计算在n×n的棋盘上放置n个皇后,使得它们互不攻击的所有可能方案数量。文章提供了Python实现代码,并分析了其时间和空间复杂度。

题目描述:
n 皇后问题 研究的是如何将 n 个皇后放置在 n × n 的棋盘上,并且使皇后彼此之间不能相互攻击。
给你一个整数 n ,返回 n 皇后问题 不同的解决方案的数量。
示例:

  1. 输入:n = 4
    输出:2
  2. 输入:n = 1
    输出:1
    思路:
    Leetcode51一样的思路,只是不需要转换形式把结果输出来,只需统计结果的个数。
class Solution(object):
    def totalNQueens(self, n):
        """
        :type n: int
        :rtype: int
        """
        self.result = []
        self.put_queens(0, [], [], [], n, [])
        return len(self.result)

    def put_queens(self, level, col, pie, na, n, cur_state):
        if level == n:
            self.result.append(cur_state)
            return
        for i in range(n):
            if i not in col and i+level not in pie and i-level not in na:
                self.put_queens(level+1, col+[i], pie+[i+level], na+[i-level], n, cur_state+[i])

复杂度分析:
时间复杂度:O(N!),其中 N是皇后数量。
空间复杂度:O(N)

LeetCode52题N皇后II要求统计n皇后不同的解决方案的数量,可使用回溯算法解决。回溯算法是一种通过尝试所有可能的解决方案,并在发现当前方案不可行时回退到上一步的算法。 ### C++解决方案 ```cpp class Solution { public: int count; int totalNQueens(int n) { count = 0; vector<int> colums(n); backtracking(n,0,colums); return count; } //回溯函数 void backtracking(int n,int row,vector<int>& colums){ //是否在所有n行都摆放好了皇后? if(row == n){ count++;//如果是的,表明找到了新的摆放方法,计数加1 return; } //尝试着将皇后放置在当前行中的每一列 for(int col=0;col<n;++col){ colums[row] = col; //检查是否合法,如果合法就继续到下一行 if(check(row,col,colums)){ backtracking(n,row+1,colums); } //如果不合法,就不要把皇后放在这列中(回溯) colums[row]=-1; } } //检查每行摆放的皇后是否合法 bool check(int row,int col,vector<int> colums){ for(int r=0;r<row;++r){//遍历每行 if(colums[r]==col || row-r==abs(colums[r]-col)){//正上方,和两侧斜对角线上不能有皇后 return false; } } return true; } }; ``` 此代码中,`totalNQueens`函数初始化计数变量`count`,并调用`backtracking`函数开始回溯过程。`backtracking`函数尝试在每一行放置皇后,若放置合法则继续到下一行,若不合法则回溯。`check`函数用于检查当前位置是否可以放置皇后,确保正上方和两侧斜对角线上没有其他皇后 [^2]。 ### C解决方案 ```c #include<stdio.h> #include<math.h> #include <stdbool.h> int rank[20];//pos列i行 bool vis[20];//标记第i行是否走过 int n=4,cnt=0; void dfs(int pos){ if(pos==n+1){//递归边界条件 cnt++; return; } for(int i=1;i<=n;i++){//枚举每行 if(vis[i]==false){ bool flag=true; for(int j=1;j<pos;j++){//枚举pos之前的皇后 if(abs(pos-j)==abs(i-rank[j])){ flag=false; break; } } if(flag){ rank[pos]=i;//pos列在i行 vis[i]=true; dfs(pos+1); vis[i]=false; } } } } int main(){ // scanf("%d",&n); dfs(1); printf("%d",cnt); return 0; } ``` 该代码使用深度优先搜索(DFS)实现回溯算法。`dfs`函数尝试在每一列放置皇后,若放置合法则标记该行已走过并递归到下一列,若不合法则回溯。`main`函数调用`dfs`函数开始搜索,并输出解决方案的数量 [^5]。
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值