从递归函数返回字典

时间:2015-03-31 20:27:09

标签: python dictionary recursion binary-tree

我有一个二叉搜索树,每个节点代表一个游戏长度。我必须返回一个字典,其中键是游戏的长度,值是该长度的游戏数。递归调用遍历树中的每个节点,但它返回一个不正确的字典。我很肯定问题是我如何返回字典。任何帮助将不胜感激

game_len = {}
if not node.children:
    key = len(node.possible_next_moves())
    if key not in game_len:
        game_len[key] = 1
    else:
        game_len[key] += 1
else:
    key = len(node.possible_next_moves())
    if key not in game_len:
        game_len[key] = 1
    else:
        game_len[key] += 1
    [game_lengths(child) for child in node.children] 
return game_len

1 个答案:

答案 0 :(得分:4)

通常,有两种方法可以处理递归算法的返回值。您可以从递归调用中收集返回值并将它们组合,也可以传入递归调用可以修改的额外可变参数。我认为后者在这种情况下可能是最好的,因为词典很容易变异,但不是特别容易合并在一起:

def game_lengths(node, result=None):
    if result is None:
        result = {}

    #... add a value to the result dict, handle base cases, etc.

    for child in node.children:
        game_lengths(child, result)

    return result