Python：二叉搜索树和递归返回嵌套列表答案

【问题标题】：Python: Binary search tree and recursion returning a nested listPython：二叉搜索树和递归返回嵌套列表
【发布时间】：2012-11-06 17:50:41
【问题描述】：

给定一个二叉搜索树，例如：

我想获取值在 a 和 b（含）之间的节点的有序列表。

我已经实现了一个递归辅助函数：

def _range(root, a, b):
node_list = []
if root.left:
    node_list.append(_range(root.left, a, b))
if root.value >= a and root.value <= b:
    node_list.append(root)
if root.right:
    node_list.append(_range(root.right, a, b))
return node_list

这给了我：

[[[BTNode: 2], BTNode: 3], BTNode: 5, [BTNode: 7, [[BTNode: 8], BTNode: 9]]]

但是我需要：

[BTNode: 2, BTNode: 3, BTNode: 5, BTNode: 7, BTNode: 8, BTNode: 9]

有没有办法让列表变平？我意识到我正在做的是将一个列表附加到另一个列表，但我不确定如何在不破坏代码的情况下解决这个问题。

【问题讨论】：

您是否尝试使用node_list.extend 而不是node_list.append？另一种解决方案是将列表作为参数传递。
我花了一段时间才弄清楚你的树是从左到右而不是从上到下绘制的。那真是令人困惑……^^

标签： python recursion tree binary-search-tree inorder

【解决方案1】：

append 将您作为参数传递的对象添加到列表中，因此您显然会得到嵌套列表。您想连接结果，因此您想使用list.extend（或+= 运算符）：

>>> from collections import namedtuple
>>> class Node(namedtuple('Node', 'value left right')):
...     # I reimplement __str__ just to avoid too much output later
...     def __repr__(self):
...             return 'Node(%r)' % self.value
...     __str__ = __repr__
... 
>>> def make_node(value, left=None, right=None):
...     return Node(value, left, right)
... 
>>> def _range(root, a, b):
...     node_list = []
...     if root.left:
...             node_list.extend(_range(root.left, a, b))
...     if a <= root.value <= b:
...             node_list.append(root)
...     if root.right:
...             node_list.extend(_range(root.right, a, b))
...     return node_list
... 
>>> tree = make_node(5, make_node(3, make_node(2)), make_node(7, None, make_node(8, None, make_node(9))))
>>> _range(tree, 1, 3)
[Node(2), Node(3)]
>>> _range(tree, 1, 10)
[Node(2), Node(3), Node(5), Node(7), Node(8), Node(9)]

另一种方法是将列表作为参数传递：

>>> def _range(root, a, b, node_list=None):
...     if node_list is None:
...             # Note: do *not* use "[]" as a default!
...             node_list = []
...     if root.left:
...             _range(root.left, a, b, node_list)
...     if a <= root.value <= b:
...             node_list.append(root)
...     if root.right:
...             _range(root.right, a, b, node_list)
...     return node_list
... 
>>> _range(tree, 1, 3)
[Node(2), Node(3)]
>>> _range(tree, 1, 10)
[Node(2), Node(3), Node(5), Node(7), Node(8), Node(9)]

【讨论】：