問題描述

操作給定的二叉樹，將其變換為源二叉樹的映象。

思路描述

1. 程式碼比文字更直觀
2. 文字描述：新建一個二叉樹，利用遞迴法，將源二叉樹上的左節點賦值到新二叉樹的右節點，將源二叉樹上的右節點賦值到新二叉樹的左節點。

Python程式碼

# 方式1：生成新的映象二叉樹
def getMirrorBST(self, root):
    if root == None:
        return
    newTree = treeNode(root.val)
    newTree.right = self.getMirrorBST(root.left)
    newTree.left = self.getMirrorBST(root.right)
    return
 
 newTree

但是提交程式碼後，說通過率為0… 原來要求將原有的二叉樹就地改成映象二叉樹…如此一來，程式碼就更簡單了：因為交換根節點的左右子節點時，以左右子節點為根節點的左子樹和右子樹也會交換位置。最終的Python程式碼如下：

# 方式2：改變給定的二叉樹為映象二叉樹
def turnToMirror(self, root):
    if root == None:
        return
    root.right, root.left = root.left, root.right
    self.turnToMirror(root.left)
    self.turnToMirror(root.right)
    return
 
 root

包含測試程式碼的最終程式碼如下：

class Solution:
    # 給定一個二叉樹，獲得其映象（軸對稱）的映象二叉樹：
    # 方式1：生成新的映象二叉樹
    def getMirrorBST(self, root):
        if root == None:
            return
        newTree = treeNode(root.val)
        newTree.right = self.getMirrorBST(root.left)
        newTree.left = self.getMirrorBST(root.right)
        return
 
 newTree
    # 方式2：改變給定的二叉樹為映象二叉樹
    def turnToMirror(self, root):
        if root == None:
            return
        root.right, root.left = root.left, root.right
        self.turnToMirror(root.left)
        self.turnToMirror(root.right)
        return root


    # 給定二叉樹的前序遍歷和中序遍歷，獲得該二叉樹
    def getBSTwithPreTin(self, pre, tin):
        if len(pre)==0 | len(tin)==0:
            return None

        root = treeNode(pre[0])
        for order,item in enumerate(tin):
            if root .val == item:
                root.left = self.getBSTwithPreTin(pre[1:order+1], tin[:order])
                root.right = self.getBSTwithPreTin(pre[order+1:], tin[order+1:])
                return root

class treeNode:
    def __init__(self, x):
        self.left = None
        self.right = None
        self.val = x

if __name__ == '__main__':
    flag = "turnToMirror"
    solution = Solution()
    preorder_seq = [1, 2, 4, 7, 3, 5, 6, 8]
    middleorder_seq = [4, 7, 2, 1, 5, 3, 8, 6]
    treeRoot1 = solution.getBSTwithPreTin(preorder_seq, middleorder_seq)
    if flag == "mirrorBST":
        newRoot = solution.getMirrorBST(treeRoot1)
        print(newRoot)
    if flag == "turnToMirror":
        solution.turnToMirror(treeRoot1)
        print(treeRoot1)