Question

我对Python和整个递归函数都很陌生，所以请原谅我的无知。

我正在尝试在Python中实现二进制搜索树，并使用以下插入方法（从类中取出）：

def insert(self, key, root=None):
    '''Inserts a node in the tree'''
    if root == None:
        root = self.root
    if root.key == None:
        self._update(root, key)
        return 0
    else:
        tmp = root
        if key > tmp.key: # we work with the right subtree
            self.insert(key, root=tmp.right)
        elif key < tmp.key: # we work with the left subtree
            self.insert(key, root=tmp.left)
        else: # key already exists
            return 0

我不确定这是否清晰，但是它会遍历树，直到它变为None值并使用要插入的键更新节点。

现在，该方法运行良好，并从头开始正确创建BST。但是返回语句存在问题，因为如果没有执行递归，它只返回0。

>>> bst.insert(10)
0
>>> bst.insert(15)
>>> bst.root.right.key
15
>>>

“插入”根键再次返回0（从第15行开始）。

>>> bst.insert(10)
0

我无法弄清楚为什么会这样。如果我在第6行放置一个print语句，它会正确执行，但它不会在第一次插入后返回任何内容。为什么是这样？（我很确定我错过了一些关于Python和递归的基本信息）

感谢您的帮助，

伊万

P.S。：我读过递归并不是实现BST的最佳方法，所以我会研究其他解决方案，但在继续之前我想知道答案。

Answer 1

在递归线上，您不会返回任何内容。如果您希望它返回0，您应该用以下行替换它们：

return self.insert(key, root=tmp.left)

而不仅仅是

self.insert(key, root=tmp.left)

Answer 2

你在一个函数里面想要返回一个值，你做什么？你写了

def function():
    return value

在你的情况下，你想要返回函数调用返回的值，所以你必须这样做。

def function():
    return another_function()

但是你做了

def function():
    another_function()

为什么你认为这应该有效？当然你使用递归，但在这种情况下，你应该记住Python的Zen，它简单地说：

特殊情况不足以打破规则。

Answer 3

在递归情况下，您需要一个return语句。尝试进行此调整。

def insert(self, key, root=None):
    '''Inserts a node in the tree'''
    if root == None:
        root = self.root
    if root.key == None:
        self._update(root, key)
        return 0
    else:
        tmp = root
        if key > tmp.key: # we work with the right subtree
            return self.insert(key, root=tmp.right)
        elif key < tmp.key: # we work with the left subtree
            return self.insert(key, root=tmp.left)
        else: # key already exists
            return 0

Python递归和返回语句

3 个答案: