【Python排序搜尋基本演算法】之二叉樹的深度和寬度

        接著上一個二叉樹的主題，用python寫一下求二叉樹深度和寬度的代碼，求深度用遞迴；求寬度用隊列，然後把每層的寬度求出來，找出最大的就是二叉樹的寬度，如下：

import queueclass Node:def __init__(self,value=None,left=None,right=None):self.value=valueself.left=leftself.right=rightdef treeDepth(tree):if tree==None:return 0leftDepth=treeDepth(tree.left)rightDepth=treeDepth(tree.right)if leftDepth>rightDepth:return leftDepth+1if rightDepth>=leftDepth:return rightDepth+1def treeWidth(tree):curwidth=1maxwidth=0q=queue.Queue()q.put(tree)while not q.empty():n=curwidthfor i in range(n):tmp=q.get()curwidth-=1if tmp.left:q.put(tmp.left)curwidth+=1if tmp.right:q.put(tmp.right)curwidth+=1if curwidth>maxwidth:maxwidth=curwidthreturn maxwidthif __name__=='__main__':root=Node('D',Node('B',Node('A'),Node('C')),Node('E',right=Node('G',Node('F'))))depth=treeDepth(root)width=treeWidth(root)print('depth:',depth)print('width:',width)

        輸出為：

depth: 4width: 3

轉載請註明：