python输出一棵树_用python实现一棵树的搜索函数

我正在尝试为n元树建立一个搜索函数。下面是节点类的用法：class node(object):

"""docstring for node"""

def __init__(self, val=''):

self.val = val #value of the node

self.subtrees = [] #list of subtree node objects

下面是我如何调用搜索函数的代码：

^{pr2}$

有一个节点的值为“1”。我希望节点对象在搜索成功后返回。在

以下是我实现的第一个搜索功能：#temp_search v0.1

def temp_search(node, key):

if node.val == key:

print 'Found', node

return node

for subtree in node.subtrees:

return temp_search(subtree, key)

上面的东西返回'None'，它从未打印'Found'

现在我修改了一下：#temp_search v0.2

def temp_search(node, key):

if node.val == key:

print 'found', node

return node

for subtree in node.subtrees:

temp_search(subtree, key)

虽然它返回'None'，