Python樹的鏡像的實現(xiàn)示例

更新時間：2023年11月23日 11:34:58 作者：Echo_Wish

樹的鏡像是指將樹的每個節(jié)點的左右子樹交換,得到一棵新的樹,本文主要介紹了Python樹的鏡像的實現(xiàn)示例,具有一定的參考價值,感興趣的可以了解一下

樹的鏡像算法

樹的鏡像可以通過遞歸遍歷樹的每個節(jié)點，交換其左右子樹來實現(xiàn)。遞歸的終止條件是遇到null節(jié)點，此時無需進行交換。

class TreeNode:
    def __init__(self, value):
        self.val = value
        self.left = None
        self.right = None

def mirror_tree(root):
    if not root:
        return None

    # 交換左右子樹
    root.left, root.right = root.right, root.left

    # 遞歸處理左右子樹
    mirror_tree(root.left)
    mirror_tree(root.right)

    return root

示例

考慮以下二叉樹：

# 構建二叉樹
"""
        1
       / \
      2   3
     / \
    4   5
"""
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)
python
Copy code
# 對樹進行鏡像處理
mirrored_tree = mirror_tree(root)

# 輸出鏡像后的樹
def print_tree(root):
    if root:
        print_tree(root.left)
        print(root.val, end=" ")
        print_tree(root.right)

print("原始樹:")
print_tree(root)
print("\n鏡像樹:")
print_tree(mirrored_tree)

輸出結果：