數(shù)據(jù)結(jié)構(gòu)(二)二叉樹的遍歷

# -*- coding: UTF-8 -*-


class Node(object):
    def __init__(self, val):
        self.__val = val
        self.__left = None  # 聲明一個成員屬性
        self.__right = None

    def set_val(self, val):
        self.__val = val

    def set_left(self, left):
        self.__left = left

    def set_right(self, right):
        self.__right = right

    def get_val(self):
        return self.__val

    def get_left(self):
        return self.__left

    def get_right(self):
        return self.__right


# 前序遍歷
def pre_order_traverse(root):
    if root is None:
        return

    print(root.get_val())
    pre_order_traverse(root.get_left())
    pre_order_traverse(root.get_right())


# 中序遍歷
def in_order_traverse(root):
    if root is None:
        return

    in_order_traverse(root.get_left())
    print(root.get_val())
    in_order_traverse(root.get_right())


# 后序遍歷
def post_order_traverse(root):
    if root is None:
        return

    post_order_traverse(root.get_left())
    post_order_traverse(root.get_right())
    print(root.get_val())


# 定義一棵二叉樹
#       1
#       /\
#      2  3
#      /\  \
#     4  5  6

node1 = Node(1)
node2 = Node(2)
node1.set_left(node2)
node3 = Node(3)
node1.set_right(node3)
node4 = Node(4)
node2.set_left(node4)
node5 = Node(5)
node2.set_right(node5)
node6 = Node(6)
node3.set_right(node6)

# 前序遍歷
print("前序遍歷")
pre_order_traverse(node1)

# 中序遍歷
print("中序遍歷")
in_order_traverse(node1)

# 后序遍歷
print("后序遍歷")
post_order_traverse(node1)
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容