本文实例讲述了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程序设计有所帮助。