您好,登錄后才能下訂單哦!
本文實例講述了Python二叉樹的鏡像轉換實現方法。分享給大家供大家參考,具體如下:
問題描述
操作給定的二叉樹,將其變換為源二叉樹的鏡像。
思路描述
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)
更多關于Python相關內容感興趣的讀者可查看本站專題:《Python數據結構與算法教程》、《Python加密解密算法與技巧總結》、《Python編碼操作技巧總結》、《Python函數使用技巧總結》、《Python字符串操作技巧匯總》及《Python入門與進階經典教程》
希望本文所述對大家Python程序設計有所幫助。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。