劍指offer:二叉樹的映象(Python)
阿新 • • 發佈:2019-02-08
問題描述
操作給定的二叉樹,將其變換為源二叉樹的映象。
思路描述
- 程式碼比文字更直觀
- 文字描述:新建一個二叉樹,利用遞迴法,將源二叉樹上的左節點賦值到新二叉樹的右節點,將源二叉樹上的右節點賦值到新二叉樹的左節點。
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)