在介紹樹的時候,我們有提到樹的相關性質,其中,樹的高度就是其中一個。
今天的目標就是來計算樹的高度。
其實只要幾行程式碼搭配遞迴思維,就可以完成!
再來就是考慮幾種狀況:
def height(root):
if root is None:
return 0
elif root.right is None and root.left is None:
return 0
else:
return 1+max(height(root.right), height(root.left))