Python回溯(皇后棋盘)

2024-05-14 11:01:29 发布

您现在位置:Python中文网/ 问答频道 /正文

好吧,经过一个又一个小时的努力,我终于明白了回溯是如何工作的。尽管我仍然认为还有一些秘密有待了解。不管怎样,我想在棋盘上制造4皇后问题。实际上,我给自己一个空白的棋盘(实际上是4x4),我需要找到4个地方,那些皇后不会攻击自己。皇后也攻击它所属的行、列和对角线。 以下是我到目前为止所做的工作(有效的func还没有完成,但这并不重要。我只需要回溯功能来工作)。你知道吗

def findNextCell(grid,i,j):

    for index in range(i,4):
        for jindex in range(j,4):
            return index, jindex

    for index in range(0,4):
        for jindex in range(0,4):
            if grid[index][jindex] == 0:
                return index, jindex

    return -1,-1


def isValid(grid, index, jindex, queen):

    rowOk = all([queen != grid[index][x] for x in range(4)])
    if rowOk:
        columnOk = all([queen != grid[x][jindex] for x in range(4)])
        if columnOk:
            return True
    return False

def queenSolve(grid, i=0, j=0, num_queens = 0):

    i, j = findNextCell(grid, i, j)
    print grid
    if num_queens == 4:
        return True
    if isValid(grid, i, j, 1):
        grid[i][j]=1
        num_queens += 1
        if (queenSolve(grid, i, j, num_queens)):
            return True
        grid[i][j]=0
    return False

input = [[0,0,0,0],[0,0,0,0],[0,0,0,0],[0,0,0,0]]

print queenSolve(input)

Tags: intrueforindexreturnifdefrange
1条回答
网友
1楼 · 发布于 2024-05-14 11:01:29

我看到的第一个问题是这个例程中的初始嵌套循环:

def findNextCell(grid,i,j):

    for index in range(i,4):
        for jindex in range(j,4):
            return index, jindex

    for index in range(0,4):
        for jindex in range(0,4):
            if grid[index][jindex] == 0:
                return index, jindex

    return -1,-1

这似乎是一些从未被清除的早期测试代码:

    for index in range(i,4):
        for jindex in range(j,4):
            return index, jindex

但不幸的是,它导致findNextCell()总是返回传递的ij,仅此而已:

>>> findNextCell(0, 1, 2)
(1, 2)
>>> findNextCell(0, 3, 2)
(3, 2)
>>> findNextCell(0, 1, 1)
(1, 1)
>>> 

我看到的下一个问题是这对线:

num_queens += 1
if (queenSolve(grid, i, j, num_queens)):

这是增加皇后的数量,成功或失败!相反,您可能只需要一行:

if (queenSolve(grid, i, j, num_queens + 1)):

这就是增加递归中皇后的数量,但在失败的情况下不影响当前的数量。你知道吗

我看到的第三个问题是,这不是一个循环:

i, j = findNextCell(grid, i, j)
...
if isValid(grid, i, j, 1):
    grid[i][j]=1
    num_queens += 1
    if (queenSolve(grid, i, j, num_queens)):
        return True
    grid[i][j]=0

您应该尝试将当前queen放在所有有效空间中,直到用完有效空间,findNextCell()返回负值,或者直到递归成功为止。你为现在的女王尝试一个位置,然后放弃。这个问题在当前皇后上是迭代的,在随后的皇后上是递归的。你知道吗

相关问题 更多 >

    热门问题