飘逸的python - 八皇后问题简洁解法

最新推荐文章于 2024-05-16 17:32:56 发布

mattkang

最新推荐文章于 2024-05-16 17:32:56 发布

阅读量1w

点赞数 2

分类专栏： Python 文章标签：八皇后 python

本文链接：https://blog.csdn.net/handsomekang/article/details/41308993

版权

Python 专栏收录该内容

128 篇文章 8 订阅

订阅专栏

思路:

使用DFS.
用一维数组表达坐标,其中下标为行,元素为列.A[i]=j表示将第i行的皇后放在j列上.
一行一行依次遍历(从上往下),决定放在哪列(从左往右),这样就不用判断行冲突,只需要判断列冲突和主斜线副斜线冲突.
(行-列)标识主斜线, (行+列)标识副斜线.

下面上代码.

#coding=utf-8

#风格1
def queen(A, cur=0):
    if cur==len(A):
        print A
    else:
        for col in range(len(A)):
            A[cur] = col #表示把第cur行的皇后放在col列上
            ok = True
            for r in range(cur):
                if A[r]==col or r-A[r]==cur-A[cur] or r+A[r]==cur+A[cur]:#判断是否跟前面的皇后冲突
                    ok = False
                    break
            if ok:
                queen(A, cur+1)

#风格2
def queen(A, cur=0):
    if cur==len(A):
        print A
    else:
        for col in range(len(A)):
            A[cur] = col #表示把第cur行的皇后放在col列上
            for r in range(cur):
                if A[r]==col or r-A[r]==cur-A[cur] or r+A[r]==cur+A[cur]:#判断是否跟前面的皇后冲突
                    break
            else:
                queen(A, cur+1)


#风格3
def queen(A, cur=0):
    if cur==len(A):
        print A
    else:
        for col in range(len(A)):
            A[cur] = col #表示把第cur行的皇后放在col列上
            if all(A[r]!=A[cur] and r-A[r]!=cur-A[cur] and r+A[r]!=cur+A[cur] for r in range(cur)):#判断是否跟前面的皇后冲突
                queen(A, cur+1)


queen([None]*8)

这三种风格差别在于" 判断是否跟前面的皇后冲突"这里.

分别是ok flag -> for...else... -> all(推导式).

孰好孰坏呢?