您的位置:首页 > 编程语言 > Python开发

Leetcode 104 python 二叉树的最大深度

2019-03-08 15:54 771 查看

题意:

给定一个二叉树,找出其最大深度(二叉树的深度为根节点到最远叶子节点的最长路径上的节点数)。
说明: 叶子节点是指没有子节点的节点。

示例:

给定二叉树 [3,9,20,null,null,15,7]

3
/ \
9  20
/  \
15   7

返回它的最大深度 3 。

方法一:

利用一个stack(栈)放置从树中遍历出来的节点,设置一个指针(gp),遇到每层最后一个往后移,层数加一;

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
def maxDepth(self, root):
"""
:type root: TreeNode
:rtype: int
"""
ans = 0
if root == None:
return ans
st = [] #栈
st.append(root)
# temp, gp = st[0], st[0]
gp = st[0] #gp为指向每层最后一个节点的指针
ans += 1 #ans表示层数
while len(st) != 0:
temp = st[0]
if temp.left != None:
st.append(temp.left)
if temp.right != None:
st.append(temp.right)
#遇到每层最后一个往后移,层数加一
if gp == temp:
gp = st[len(st)-1]
if gp != st[0]:
ans += 1
st.pop(0)
return ans

方法二:递归

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
def maxDepth(self, root):
"""
:type root: TreeNode
:rtype: int
"""

if root == None:
return 0
else:
return max(self.maxDepth(root.left),self.maxDepth(root.right))+1
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: