題目一:
輸入一棵二叉樹,求該樹的深度。從根結(jié)點(diǎn)到葉結(jié)點(diǎn)依次經(jīng)過的結(jié)點(diǎn)(含根、葉結(jié)點(diǎn))形成樹的一條路徑,最長(zhǎng)路徑的長(zhǎng)度為樹的深度。
思路:
- 如果樹只有一個(gè)節(jié)點(diǎn),那么深度為1
- 如果樹的根節(jié)點(diǎn)只有左子樹,沒有右子樹,那么深度等于左子樹的深度+1
- 如果樹的根節(jié)點(diǎn)只有右子樹,沒有左子樹,那么深度等于右子樹的深度+1
- 如果樹的根節(jié)點(diǎn)既有左子樹又有右子樹,那么深度等于max(left,right)+ 1
# -*- coding:utf-8 -*-
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def TreeDepth(self, pRoot):
# write code here
if pRoot is None:
return 0
left = self.TreeDepth(pRoot.left)
right = self.TreeDepth(pRoot.right)
return max(left,right)+1
題目二
輸入一棵二叉樹,判斷該二叉樹是否是平衡二叉樹。如果某二叉樹中任意節(jié)點(diǎn)的左右子樹的深度相差不超過1,那么它是一顆平衡二叉樹。
思路一:
采用上一題計(jì)算深度的思路,調(diào)用TreeDepth得到左右子樹的深度,判斷其差值是否大于1 。
# -*- coding:utf-8 -*-
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def TreeDepth(self, pRoot):
# write code here
if pRoot is None:
return 0
left = self.TreeDepth(pRoot.left)
right = self.TreeDepth(pRoot.right)
return max(left,right)+1
def IsBalanced_Solution(self, pRoot):
# write code here
if pRoot is None:
return True
left = self.TreeDepth(pRoot.left)
right = self.TreeDepth(pRoot.right)
diff = abs(left-right)
if diff > 1:
return False
return self.IsBalanced_Solution(pRoot.left) and self.IsBalanced_Solution(pRoot.right)