用Python實(shí)現(xiàn)二叉樹(shù)結(jié)構(gòu)及三種遍歷

本篇內(nèi)容介紹了“用Python實(shí)現(xiàn)二叉樹(shù)結(jié)構(gòu)及三種遍歷”的有關(guān)知識(shí)，在實(shí)際案例的操作過(guò)程中，不少人都會(huì)遇到這樣的困境，接下來(lái)就讓小編帶領(lǐng)大家學(xué)習(xí)一下如何處理這些情況吧！希望大家仔細(xì)閱讀，能夠?qū)W有所成！

一：代碼實(shí)現(xiàn)

class TreeNode:
    """節(jié)點(diǎn)類(lèi)"""
    def __init__(self, mid, left=None, right=None):
        self.mid = mid
        self.left = left
        self.right = right


# 樹(shù)類(lèi)
class Tree:
    """樹(shù)類(lèi)"""
    def __init__(self, root=None):
        self.root = root

    def add(self, item):
        # 將要添加的數(shù)據(jù)封裝成一個(gè)node結(jié)點(diǎn)
        node = TreeNode(item)
        if not self.root:
            self.root = node
            return
        queue = [self.root]
        while queue:
            cur = queue.pop(0)
            if not cur.left:
                cur.left = node
                return
            else:
                queue.append(cur.left)

            if not cur.right:
                cur.right = node
                return
            else:
                queue.append(cur.right)
               
tree = Tree()
tree.add(0)
tree.add(1)
tree.add(2)
tree.add(3)
tree.add(4)
tree.add(5)
tree.add(6)

二：遍歷

在上述樹(shù)類(lèi)代碼基礎(chǔ)上加遍歷函數(shù)，基于遞歸實(shí)現(xiàn)。

先序遍歷：

先序遍歷結(jié)果是：0 -> 1 -> 3 -> 4 -> 2 -> 5 -> 6

# 先序遍歷
    def preorder(self, root, result=[]):
        if not root:
            return
        result.append(root.mid)
        self.preorder(root.left, result)
        self.preorder(root.right, result)
        return result
        
print("先序遍歷")
print(tree.preorder(tree.root))
"""
先序遍歷
[0, 1, 3, 4, 2, 5, 6]
"""

中序遍歷：

中序遍歷結(jié)果是：3 -> 1 -> 4 -> 0 -> 5 -> 2 -> 6

# 中序遍歷
    def inorder(self, root, result=[]):
        if not root:
            return result
        self.inorder(root.left, result)
        result.append(root.mid)
        self.inorder(root.right, result)
        return result
        
print("中序遍歷")
print(tree.inorder(tree.root))
"""
中序遍歷
3, 1, 4, 0, 5, 2, 6]
"""

后續(xù)遍歷

后序遍歷結(jié)果是：3 -> 4 -> 1 -> 5 -> 6 -> 2 -> 0

# 后序遍歷
    def postorder(self, root, result=[]):
        if not root:
            return result
        self.postorder(root.left, result)
        self.postorder(root.right, result)
        result.append(root.mid)

        return result
        
print("后序遍歷")
print(tree.postorder(tree.root))
"""
后序遍歷
[3, 4, 1, 5, 6, 2, 0]
"""

“用Python實(shí)現(xiàn)二叉樹(shù)結(jié)構(gòu)及三種遍歷”的內(nèi)容就介紹到這里了，感謝大家的閱讀。如果想了解更多行業(yè)相關(guān)的知識(shí)可以關(guān)注億速云網(wǎng)站，小編將為大家輸出更多高質(zhì)量的實(shí)用文章！