Python回溯(女王检查板)

时间:2016-05-26 18:04:21

标签: python backtracking

好吧,经过几个小时的挣扎,我终于明白了回溯是如何运作的。虽然我仍然认为还有秘密可以理解。无论如何,我正试图在棋盘上创建4皇后问题。实际上,我给自己一个空白的棋盘(实际上是4x4),我需要找到4个不会攻击自己的地方。女王也攻击它所属的行,列和对角线。 这是我到目前为止所做的(有效的函数还没有完成,但它并不重要。我只需要回溯函数来工作)。

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)

1 个答案:

答案 0 :(得分:1)

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

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

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