递归计算访问的节点



我需要计算我在寻找令牌时访问过树上的节点数(在本例中为 1)。到目前为止,我已经想出了:

def search_recursive(node,rounds=0):
    if len(node)==2:
         if node[1]==1:
             print rounds
    else:
         rounds += 1
         huff_search_recursive(node[1],rounds)
         huff_search_recursive(node[2],rounds)

但是,如果我尝试将回合分配给以下任何内容:

def search_recursive(node,rounds=0):
    if len(node)==2:
        if node[1]==1:
            return rounds
    else:
        rounds += 1
        huff_search_recursive(node[1],rounds)
        huff_search_recursive(node[2],rounds)

Python 将 None 赋给任何变量。例如。

rounds = search_recursive(node)

节点的示例如下:

node = (1.0, (0.5, 0), (0.5, (0.25, 0), (0.25, (0.125, 1), (0.125, 0))))

谁能向我解释导致此错误的原因,以及我将来如何测试它?

在每个

函数调用中,rounds是一个不同的变量。更改它不会更改外部函数调用的 rounds 变量。您需要返回值并使用返回的值。

def search_recursive(node):
    rounds = 0
    if len(node)==2:
        if node[1]==1:
            return rounds
    else:
        rounds += 1
        rounds += search_recursive(node[1])
        rounds += search_recursive(node[2])
    return rounds

最新更新