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

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

作者: 机智的柠檬 | 来源:发表于2019-10-12 18:28 被阅读0次

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

  • 如果一个二叉树是空的,那么其高度为0;
  • 如果一个二叉树的只有一个节点,那么其高度为1;
  • 对于非叶子节点的二叉树,其高度为左子树与右子树高度的最大值
    二叉树的高度可以用递归算法来实现:
  • 如果输入的是空节点,高度为0;
  • 如果输入的是叶子节点,左右子树都为空,则其高度为1;
  • 如果输入的是非叶子节点,那么分别计算左右子树的高度,选取最大值加1,即为二叉树的高度。
    树的节点实现
    定义二叉树节点的数据结构,其包含value以及左孩子与右孩子
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

相关文章

  • 二叉树2-平衡二叉树、完全二叉树、二叉树剪枝

    110.平衡二叉树 给定一个二叉树,判断它是否是高度平衡的二叉树。 一棵高度平衡二叉树定义为:一个二叉树每个节点 ...

  • 力扣算法 - 平衡二叉树

    平衡二叉树 给定一个二叉树,判断它是否是高度平衡的二叉树。本题中,一棵高度平衡二叉树定义为:一个二叉树每个节点 的...

  • 110. 平衡二叉树

    题目描述 给定一个二叉树,判断它是否是高度平衡的二叉树。 本题中,一棵高度平衡二叉树定义为: 一个二叉树每个节点 ...

  • Leetcode 110. 平衡二叉树

    题目描述 给定一个二叉树,判断它是否是高度平衡的二叉树。 本题中,一棵高度平衡二叉树定义为: 一个二叉树每个节点 ...

  • 平衡二叉树

    题目描述:给定一个二叉树,判断它是否是高度平衡的二叉树。 本题中,一棵高度平衡二叉树定义为: 一个二叉树每个节点 ...

  • leetcode--110--平衡二叉树

    题目:给定一个二叉树,判断它是否是高度平衡的二叉树。 本题中,一棵高度平衡二叉树定义为: 一个二叉树每个节点 的左...

  • leetcode-二叉树-平衡二叉树(110)

    题目 给定一个二叉树,判断它是否是高度平衡的二叉树。 本题中,一棵高度平衡二叉树定义为: 一个二叉树每个节点 的左...

  • Leetcode.101.Symmetric Tree

    题目 给定一个二叉树, 判断这个二叉树是否对称. 思路 判断这个数是否对称: 将根节点的右边子树所有左右节点都交换...

  • 26平衡二叉树

    给定一个二叉树,判断它是否是高度平衡的二叉树。本题中,一棵高度平衡二叉树定义为:一个二叉树每个节点 的左右两个子树...

  • T110、平衡二叉树

    给定一个二叉树,判断它是否是高度平衡的二叉树。本题中,一棵高度平衡二叉树定义为:一个二叉树每个节点 的左右两个子树...

网友评论

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

      本文链接:https://www.haomeiwen.com/subject/siccmctx.html