python实现:给定一个二叉树的根节点,判断该树是否为平衡树

2019-10-12  本文已影响0人  机智的柠檬

平衡树:二叉树的每个节点的左子树和右子树的高度差不超过1.

class TreeNode:
    def __init__(self, v):
        self.value = v
        self.left = self.right = None

排序树的实现:
构建排序二叉树,创建TreeUtil类,当加入节点的值比当前节点小,则进入左子树,当加入节点值比当前节点大,则进入右子数。
向排序树中添加叶子节点函数为 addTreeNode(self, node)

class TreeUtil:
    def __init__(self):
        self.root = None
    def addTreeNode(self, node):
        if self.root = None:
            self.root = node
        currentNode = self.root
        prevNode = self.root
        while currentNode is not None:
            prevNode = currentNode
            if currentNode.value > node.value:
                currentNode = currentNode.left
            else:
                currentNode = currentNode.right
            if prevNode.value > node.value:
                prevNode.left = node
            else:
                prevNode.right = node
    def getTreeRoot(self):
        return self.root

接着递归查询二叉树中每一个的高度,如果左右子树的高度差超过1,那么二叉树就是不平衡的。
computeTreeHeight()是递归计算节点的左右子树的高度,当传入的节点是None时,返回高度0,否则递归调用自己去计算该节点的左右子树的高度。

class BalancedTree:
    def __init__(self):
        self.balanced = True
    def isTreeBalanced(self, node):
        self.computeTreeHeight(node)
        return self.balanced
    def computeTreeHeight(node):
        if node is null:
            return 0
        leftHeight = computeTreeHeight(node.left)
        rightHeight = computeTreeHeight(node.right)
        if abs(leftHeight - rightHeight) > 1:
            self.balanced = False
        height = 0
        if leftHeight > rightHeight :
            height = leftHeight
        else:
            height = rightHeight
        print("node value:{0}, left height:{1}, right height:{2}, height{}".format(node.value,leftHeight,rightHeight,height+1))
        return height + 1

构造一棵二叉树,调用上面代码判断二叉树的平衡性:

array = [6,4,9,2,5,7,10,1,3,8]
util = TreeUtil()
for node in array:
    n = TreeNode(node)
    util.addTreeNode(n)
root = util.getTreeRoot()
bt = BalancedTree()
isBalanced = bt.isTreeBalanced(root)
if isBalanced:
    print(" the binary tree is balaced")
else:
    print(" the binary tree is not balanced") 

运行结果为:
二叉树为:


image.png image.png
上一篇 下一篇

猜你喜欢

热点阅读