Return 来自递归函数的字典

Question

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

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

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

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