1. 程式人生 > >python實現二叉樹深度遍歷

python實現二叉樹深度遍歷

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)

4、輸出結果