力扣刷题(python)50天——第二十七天:二叉树的最大深度
题目描述
给定一个二叉树,找出其最大深度。
二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。
说明: 叶子节点是指没有子节点的节点。
示例:
给定二叉树 [3,9,20,null,null,15,7]
返回它的最大深度 3 。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-depth-of-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
方法
拿到后第一反应就是递归,遇到的困难是怎么让递归函数知道自己是在第几层,于是设置了depth变量。
解答
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def maxDepth(self, root):
"""
:type root: TreeNode
:rtype: int
"""
if root==None:
return 0
self.maxdepth=1
depth=1
o_depth=depth
if root.left:
depth+=1
self.findleft(depth,root.left)
if root.right:
o_depth+=1
self.findright(o_depth,root.right)
return self.maxdepth
def findleft(self,depth,root):
if depth>self.maxdepth:
self.maxdepth=depth
o_depth=depth
if root.left:
depth+=1
self.findleft(depth,root.left)
if root.right:
o_depth+=1
self.findright(o_depth,root.right)
def findright(self,depth,root):
if depth>self.maxdepth:
self.maxdepth=depth
o_depth=depth
if root.left:
depth+=1
self.findleft(depth,root.left)
if root.right:
o_depth+=1
self.findright(o_depth,root.right)
执行结果
心得:
二叉树原来可以这样在python中表示。
改进:
class Solution:
def maxDepth(self, root):
"""
:type root: TreeNode
:rtype: int
"""
if root is None:
return 0
else:
left_height = self.maxDepth(root.left)
right_height = self.maxDepth(root.right)
return max(left_height, right_height) + 1
作者:LeetCode
链接:https://leetcode-cn.com/problems/maximum-depth-of-binary-tree/solution/er-cha-shu-de-zui-da-shen-du-by-leetcode/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
这样的递归更方便,每次答案的递归都写得比我简单多了。。。