递归到迭代 - AVL 树 - isBalanced答案

【问题标题】：Recursive To Iteratative - AVL Tree - isBalanced递归到迭代 - AVL 树 - isBalanced
【发布时间】：2014-05-27 01:14:41
【问题描述】：

我必须编写一个迭代算法来确定 AVL-Tree 是否平衡。

我的第一个方法是找到直接的方法，但几个小时后我放弃了，所以我编写了递归算法并尝试对其进行转换。

这是递归版本的源代码（用python编写）

def isBalanced(root):
    if root == None:
        return -1

    lh = isBalanced(root.left)
    rh = isBalanced(root.right)

    if lh == -2 or rh == -2:
        return -2

    if abs(lh - rh) > 1:
        return -2

    return max(lh, rh) + 1

我现在的问题是，我无法转换它，也许你们中的一个可以给我一个提示或解决我的问题

提前致谢

【问题讨论】：

您需要从叶子开始，因为递归所做的是向下到叶子并将子树的高度传播到根。

标签： python algorithm recursion avl-tree iteration

【解决方案1】：

请记住，递归程序使用调用堆栈。您可以使用堆栈将任何递归程序转换为迭代程序。在下面的代码中，我使用了两个。

def isBalanced(root):
    nodes = [root]
    results = []
    while nodes:
        node = nodes.pop()
        if node is None:
            results.append(-1)
        else if node == 0: # 0 is a flag we use for when we are ready to combine results
            lh = results.pop()
            rh = results.pop()
            if abs(lh - rh) > 1:
                return -2  # we could have continued with the stack; this is just a shortcut
            else:
                results.append(max(lh, rh) + 1)
        else:
            nodes.push(0)
            nodes.push(node.left)
            nodes.push(node.right)
    return results, # results will have only one value

这里，stack 是要检查的节点堆栈，以及这些节点的结果。

【讨论】：