Python实现二叉树的深度

最新推荐文章于 2025-09-16 19:43:21 发布

原创最新推荐文章于 2025-09-16 19:43:21 发布 · 3.7k 阅读

1 ·

CC 4.0 BY-SA版权

Python 专栏收录该内容

15 篇文章

订阅专栏

本文介绍了一个使用Python实现的二叉树深度搜索算法。通过定义TreeNode类来构造二叉树节点，并实现了Solution类用于判断二叉树是否平衡。通过递归方式计算左右子树的高度并返回最大高度加一。

python实现二叉树的深度搜索

class TreeNode(object):
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None


class Solution(object):
    def isBalanced(self, root):
        if root==None:
            return 0
        leftheight=self.isBalanced(root.left)
        rightheight=self.isBalanced(root.right)
        if leftheight>=rightheight:
            return leftheight+1
        else:
            return rightheight+1



input_3=TreeNode(3)
input_4=TreeNode(4)
input_5 = TreeNode(5)
input_5.left=input_3
input_5.right=input_4
input_18 = TreeNode(18)
input_all = TreeNode(2)
input_all.left = input_5
input_all.right = input_18
slu_ = Solution()
print input_all
t = slu_.isBalanced(input_all)
print t