​LeetCode刷题实战52:N皇后 II

共 3645字,需浏览 8分钟

 ·

2020-10-01 03:15

算法的重要性,我就不多说了吧,想去大厂,就必须要经过基础知识和业务逻辑面试+算法面试。所以,为了提高大家的算法能力,这个公众号后续每天带大家做一道算法题,题目就从LeetCode上面选 !

今天和大家聊的问题叫做 N皇后 II,我们先来看题面:

https://leetcode-cn.com/problems/n-queens-ii/

The n-queens puzzle is the problem of placing n queens on an n×n chessboard such that no two queens attack each other.

题意


n 皇后问题研究的是如何将 n 个皇后放置在 n×n 的棋盘上,并且使皇后彼此之间不能相互攻击。


上图为 8 皇后问题的一种解法。

给定一个整数 n,返回 n 皇后不同的解决方案的数量。

提示:皇后,是国际象棋中的棋子,意味着国王的妻子。皇后只做一件事,那就是“吃子”。当她遇见可以吃的棋子时,就迅速冲上去吃掉棋子。当然,她横、竖、斜都可走一或 N-1 步,可进可退。

样例

输入: 4
输出: 2

解释: 4 皇后问题存在如下两个不同的解法。
[
 [".Q..",  // 解法 1
  "...Q",
  "Q...",
  "..Q."],

 ["..Q.",  // 解法 2
  "Q...",
  "...Q",
  ".Q.."]
]



解题

https://www.cnblogs.com/techflow/p/12782020.html

思路分析

八皇后问题已经是老生常谈了,在我们探讨解法之前,先来思考一个问题,用递归或者搜索解决的问题很多,为什么只有N皇后问题如此经典呢?
是因为国际象棋比较流行吗?还是因为这个问题很困难吗?还是它的思路很巧妙吗?由于这个问题是我自己提出的,书本上并没有相关的答案,需要我们自己思考。我们先不公布答案,带着这个问题来分析一下这道题的思路。
我个人在解题的时候喜欢问问题,很多时候看似破朔迷离找不到头绪的题目,多问几个问题也许就能找到灵感。如果我们对数字敏感的话,很容易发现一个大问题,为什么题目会让我们在n*n的棋盘上摆放n个皇后呢?为什么是n个,而不是2n个,也不是n+1个呢?
这个问题不难回答,因为题目当中规定了皇后不能同行摆放,所以每一行最多摆放一个皇后,一共有n行,那么显然最多只能摆放n个皇后。但是如果我们继续提问,既然这么多限制条件,那为什么一定能找到答案呢,会不会摆放n个皇后的解不存在呢?这当然是有可能的,我们很容易发现,当n=2和3的时候就没有解法。
如果我们顺着这个思路问下去,还可以挖掘出许多问题来。比如到底什么样的n可以使得一定有解呢?每一个n对应的解有没有规律呢?这样一直问下去,如果所有的问题都能解答,就说明这个问题就彻底吃透了。实际上这个问题背后是一个非常复杂的数学问题,会在之后开一篇文章单独讲解。
虽然不去深究这些问题,也一样可以把题目做出来,但很多时候思维和能力的差距就体现在这些看似无用的思考上。
我们先把问题简化,把解的存在性问题先放一放,既然题目要我们求,一定会给我们有解的n。而且我们也可以大概猜测得出结论,当n大于等于4的时候,N皇后问题一定有解。那么,我们怎么求解呢?

问题建模

我们干想是没有结果的,要对问题进行建模。建模这个词很玄乎,听起来很高端,而且在很多场景当中都会出现。比如机器学习当中,我们需要建模,还有专门的数学建模大赛,但是少有人会对建模这个词进行解释。
经过了一系列思考,我个人总结,所谓的建模,其实就是一个寻找和设计适应问题的解法的过程。模型就是从问题当中抽象出来的逻辑,比如N皇后摆放是问题本身,但是摆放的方法的逻辑才是模型。模型不是凭空出现的,是我们一点点构建的。这个过程有点像是搭积木,从无到有,从易到难,一点点将模型完善。
n*n的棋盘上摆放n个皇后,这个是问题本身,我们做第一层抽象。显然,由于皇后之间不能同行也不能同列,那么每一行和每一列只能摆放一个皇后。我们不能同时枚举一个皇后摆放的行和列,我们优先考虑其中的行。不如做一个假设,由于皇后之间没有差别,我们可以假设每一行摆放的皇后是固定的。第一个皇后就摆放在第一行,第二个皇后就摆放在第二行。
进行了第一层抽象之后,问题清晰了许多,但是还是无法得出答案。所以我们还需要做第二层抽象和分析,每行固定一个皇后之后可以保证皇后之间不会同行发生冲突,但是不能保证不同列以及不同对角线。所以我们必须设计一个机制,来保证这一点。我们需要枚举皇后所有摆放的情况,所以不能再固定皇后摆放的列,既然不能固定,但是可以记录。由于我们已经确定了每一个皇后摆放的行,只要记录下它们摆放的列,就可以判断是否会构成同列以及同对角线。
到这里,我们已经找到解法了,但是我们还可以再做第三层抽象。由于皇后已经固定了行号,我们可以用数组当中的下标代替皇后。下标0存储的位置就是皇后0摆放的列号,0就是皇后0的行号,那么我们用一个一维数组就存储了皇后摆放的二维信息。
也就是说我们在递归的时候,只用一个数组就记录了整个棋盘的情况,这个时候用代码实现起来就要容易很多。
# code for leetcode 51
class Solution:
    
    def dfs(self, n, queen, ret):
        if len(queen) == n:
            ret.append(queen[:])
            return 
        
        for i in range(n):
            # 如果同列
            if i in queen:
                continue
            flag = False
            # 判断是否存在同对角线
            for j, idx in enumerate(queen):
                # len(queen)表示当前是第几个皇后
                if abs(len(queen) - j) == abs(idx - i):
                    flag = True
                    break
            if flag:
                continue
            # 合法则放入i列
            queen.append(i)
            self.dfs(n, queen, ret)
            queen.pop()
            
    def transform(self, n, ret):
        res = []
        # 根据每个皇后摆放的列号还原棋盘
        for arr in ret:
            s = []
            for i, idx in enumerate(arr):
                row = ['.' for _ in range(n)]
                row[idx] = 'Q'
                s.append(''.join(row))
            res.append(s)
        return res
        
    def solveNQueens(self, n: int) -> List[List[str]]:
        ret = []
        self.dfs(n, [], ret)
        return self.transform(n, ret)

总结

最后,我们再回到一开始的问题,为什么递归求解的问题这么多,只有N皇后成为经典呢?
我觉得很重要的一个原因就在于这道题对应的建模过程,我们从无到有,抽丝剥茧,一点点将整个问题搭建起来,构建出了一个适配于当前问题的模型。并且经过我们的优化,这也是用递归实现的最佳模型。对于我们而言,把问题AC了其实并不重要,重要的是能够掌握这个思路构建的能力,这样以后我们就可以很方便地迁移到其他的问题场景当中,这才是学习的精髓。

好了,今天的文章就到这里,如果觉得有所收获,请顺手点个在看或者转发吧,你们的支持是我最大的动力。


上期推文:


LeetCode1-50题汇总,速度收藏!
LeetCode刷题实战51:N 皇后


浏览 48
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报