本文共 802 字,大约阅读时间需要 2 分钟。
【题目】:
输入一颗二叉树的根节点,求该二叉树的深度。二叉树的深度是从根节点到叶子节点所形成的最长的路径。
【解题思路】:
递归,分别求取左子树和右子数的深度,然后将最大的值加1进行返回。
示例代码:
# define a tree nodeclass TreeNode(object): def __init__(self, val): self.val = val self.left = None self.right = Nonedef tree_deep(root): if not root: return 0 # 这儿不能return None, 否则不能参与下面return+1的计算 left = tree_deep(root.left) right = tree_deep(root.right) return max(left, right) + 1# test# creat a treedef creat_tree(nums, index): if not nums: return None if index >= len(nums): return None root = TreeNode(nums[index]) root.left = creat_tree(nums, index * 2 + 1) root.right = creat_tree(nums, index * 2 + 2) return rootroot = creat_tree([1, 2, 3, None, 5, None, 6, None, None, 7], 0)print(tree_deep(root))
运行结果:
转载地址:http://lrhh.baihongyu.com/