递归计数访问的节点

Question

我需要计算我在树上访问的节点数量，同时查找令牌（在本例中为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))))

有人可以向我解释是什么导致了这个错误，以及我将来如何测试它？

Answer 1

在每个函数调用中，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