Question

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

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

Answer 1

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

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

从递归函数返回字典

1 个答案: