2

这是来自https://wiki.python.org/moin/SimplePrograms的 8 个皇后问题的递归解决方案。我很难理解解决函数中的 return 语句如何/为什么起作用。乍一看，它似乎违反了规则（例如，解决方案尚未在 return 语句中的循环之前声明或分配值，但它出现在 return 语句中的循环之前）但我可以成功运行它，所以我好奇地想了解它是如何工作的，以及为什么有人可能会选择这样写（混淆？简短？因为我还不明白的其他限制？）

BOARD_SIZE = 8

def under_attack(col, queens):
    left = right = col

    for r, c in reversed(queens):
        left, right = left - 1, right + 1

        if c in (left, col, right):
            return True
    return False

def solve(n):
    if n == 0:
        return [[]]

    smaller_solutions = solve(n - 1)

    return [solution+[(n,i+1)]
        for i in xrange(BOARD_SIZE)
            for solution in smaller_solutions
                if not under_attack(i+1, solution)]
for answer in solve(BOARD_SIZE):
    print answer

我熟悉一些实验和我上过的课程的问题的递归解决方案，但我对 python 整体还是很陌生（我主要在课堂上和我自己使用过 java 和 C）。

有没有人有一个很好的方法来解释这种语法是如何工作的，或者我可以研究的其他例子可能有助于我理解它？这真的勾起了我的好奇心……

4

2 回答 2

5

我猜你的困惑是这行代码：

       v---- this starts the list
return [solution+[(n,i+1)]
        for i in xrange(BOARD_SIZE)
            for solution in smaller_solutions
                if not under_attack(i+1, solution)]
                                                  ^-- this ends the list

这实际上是一个列表推导，一种创建列表的快捷方式。

这是该循环的简写版本。

x = []
for i in xrange(BOARD_SIZE):
    for solution in smaller_solutions:
        if not under_attack(i+1, solution):
            x.append(solution+[(n, i+1)])
return x
于 2013-09-24T08:04:08.683 回答
3

您在询问列表推导。这是 Python 执行功能映射（将函数应用于列表的每个元素）和过滤器（在特定条件下过滤列表）的方式。

解释这一点的最简单方法是举一些例子。

说：

>>> l = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
>>> [2 * i for i in l]
[2, 4, 6, 8, 10, 12, 14, 16, 18, 20]
>>> [i for i in l if i > 5]
[6, 7, 8, 9, 10]

现在这些可以组合起来，你可以使用任意数量的 for：

>>> # squares of even numbers below 10
>>> [i * i for i in range(10) if i%2 == 0]
[0, 4, 16, 36, 64]
>>> # concatenating all lists
>>> ll = [[1, 2], [3, 4], [5, 6], [7, 8], [8, 10]]
>>> [i for l in ll for i in l]
[1, 2, 3, 4, 5, 6, 7, 8, 8, 10]
于 2013-09-24T08:03:08.223 回答