八皇后问题

leetcode # 51 N皇后问题

本质上是一个全排列问题

class Solution(object):
    def solveNQueens(self, n):
        if n <= 0:
            return ''
        # 肯定是一行一个皇后,一列也是一个皇后
        # 所以可以定义一个一维数组,索引是行号,值是列号。比如nums[0] = 1 表示第一个皇后在第行第列
        nums = [-1] * n
        res = []
        self.Permutation(nums, 0, res, [])
        return res 
    def Permutation(self, nums, index, res, path):
        # 进行N皇后的全排列
        if index == len(nums):
            res.append(path)
            return 
        else:
            for i in range(len(nums)):
                nums[index] = i
                if self.isValid(nums, index):
                    # 第i个皇后的位置有效,进行下一个皇后的排位置
                    tmp = '.' * len(nums)
                    self.Permutation(nums, index+1, res, path + [tmp[:i]+'Q'+tmp[i+1:]]) 
                    
                
    def isValid(self, nums, n):
        # 判断皇后位置是否有效。
        # 因为一维数组已经保证N皇后们不会同行同列,所以只需要保证不再同一对角线就行了
        # 对角线也就是一条有截距的直线,y= x+b, 也就是y-x = b(是定值)
        # 只要保证 y1-y2 != x1-x2 就是有效的
        for i in range(n):
            if abs(nums[i] - nums[n]) == n-i or nums[i] == nums[n]:
                return False
        return True
        
Solution().solveNQueens(4)
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值