python實現二叉樹深度遍歷
阿新 • • 發佈:2018-11-10
1、什麼是深度優先遍歷
其實深度優先遍歷你可以把它看成是前序遍歷,比如對於如下二叉樹:
其深度遍歷的結果是:1,2,4,8,9,5,3,6,7
2、解題思路
一、採用遞迴的方式進行遍歷。
二、利用棧,先將根入棧,再將根出棧,並將根的右子樹,左子樹存入棧,按照棧的先進後出規則來實現深度優先遍歷。
3、程式設計實現
class Node(): # 節點類 def __init__(self, data=-1): self.data = data self.left = None self.right = None class Tree(): # 樹類 def __init__(self): self.root = Node() def add(self, data): # 為樹加入節點 node = Node(data) if self.root.data == -1: # 如果樹為空,就對根節點賦值 self.root = node else: myQueue = [] treeNode = self.root myQueue.append(treeNode) while myQueue: # 對已有的節點進行層次遍歷 treeNode = myQueue.pop(0) if not treeNode.left: treeNode.left = node return elif not treeNode.right: treeNode.right = node return else: myQueue.append(treeNode.left) myQueue.append(treeNode.right) def DFS(self,root): #遞迴實現深度優先遍歷 if root == None: return print(root.data) self.DFS(root.left) self.DFS(root.right) def DFS_STACK(self,root): #基於棧資料結構實現的深度遍歷 if root == None: return stack = [] stack.append(root) while stack: now_node = stack.pop() print(now_node.data) if now_node.right != None: stack.append(now_node.right) if now_node.left != None: stack.append(now_node.left) if __name__ == '__main__': # 主函式 datas = [1,2, 3, 4, 5, 6, 7, 8, 9] tree = Tree() # 新建一個樹物件 for data in datas: tree.add(data) # 逐個加入樹的節點 print('遞迴實現前序遍歷:') tree.DFS_STACK(tree.root)