在for循环中递归运行for循环(python)

时间:2016-03-30 18:46:35

标签: python loops for-loop recursion

我是python的新手,并且有想用它来解决电视节目倒计时中的数字游戏。 (rules for those unfamiliar)。我用谷歌搜索结果this has been done before - 但我没有正确理解代码,然后想到为什么不自己试一试。我已经搜索过,还有其他人正在寻找递归解决方案,但我无法让他们为我的例子工作(道歉,毕竟我对此很新)。

我想要做的是获取一个数字列表,然后循环应用操作到它们的对,并用输出替换该对。这将以递归方式重复,直到找到我们要查找的数字,或者将数字列表缩小为1。

我的功能" single_move_generator"是一个生成形式元组的生成器((a,b,操作),答案,剩下使用的数字)。我想将元组的最后一部分作为新列表反馈到函数中,但也要跟踪第一部分,因为它是历史记录'我们如何实现我们的答案。目前我有以下内容:

target = 155
numbers_to_use = [6, 25, 3, 2]
for a in single_move_generator(numbers):
    if a[1] == target:
        print(a[0])
    for b in single_move_generator(a[2]):
        if b[1] == target:
                print(a[0],b[0])
                quit()
        for c in single_move_generator(b[2]):
                if c[1] == target:
                    print(a[0],b[0],c[0])
                    quit()

产生

(25, 6, 'add') (3, 2, 'add') (31, 5, 'multiply')

但是我想能够给它一个更大的数字列表并让它继续直到列表达到第一个。我怀疑我需要一个while循环 - 但这种尝试并不起作用。它没有找到目标或跟踪移动的历史。

numbers_available = numbers
while len(numbers_available) >1 and target not in numbers_available:

    for a in single_move_generator(numbers_available):
        if a[1] == target:
            print("Target Found", a)           
            break

    numbers_available = a[2]

numbers_available = a[2]

我觉得必须有一种pythonic方式来做这个比我做得更整洁的方法 - 任何提示都会非常感激。谢谢!

2 个答案:

答案 0 :(得分:1)

根据您使用元组(i, j, operation)的想法,我写了以下内容。这是一个递归解决方案,因为main函数会自行调用它。

from itertools import combinations, product

def check_operation(i, j, operation):
    """
    Check whether 'i operation j' is permitted.
    """
    if operation == '/' and j == 0:
        return False
    elif operation == '/' and i%j != 0:
        return False
    # if not playing with negative values
    #elif operation == '-' and i-j < 0:
    #    return False
    else:
        return True

def countdown(target, numbers, trackback):
    if target in numbers:
        print trackback
    for possibility in product(combinations(numbers,2), ['+', '*', '/', '-']):
        new_numbers = [k for k in numbers] # copy list, use list.copy in python 3
        i, j = possibility[0][0], possibility[0][1]
        operation = possibility[1]
        new_numbers.remove(i)
        new_numbers.remove(j)
        if not check_operation(i, j, operation):
            continue
        new_numbers.append(eval('%i%s%i' % (i, operation, j)))
        countdown(target, new_numbers, trackback+[possibility])

countdown(155, [6, 25, 3, 2], [])

只有在存在解决方案时它才有效,因为它不会打算尽可能接近解决方案。然而,它将返回所有解决方案,而不仅仅是一个。

答案 1 :(得分:0)

根据您发布的内容,这应该适合您:

def solve(numbers, target):
  for op, res, nums in single_move_generator(numbers):
    if res == target:
      return [op]
    ops = solve(nums, target)
    if ops:
      return [op] + ops

print(*solve(numbers_to_use, target))

它应该等同于您发布的嵌套for循环。

res == target时会触发递归的底部。默认情况下,Python函数返回None,因此如果对solve的递归调用返回了一个真值,那么它必须已经达到了目标。 ops将包含最后一个操作,如果它是递归的底部。然后将其附加到启动递归调用并返回到较高级别的操作。因此该函数将返回顶层的所有操作。