剑指offer:二叉树的镜像(Python)
2018-03-07 16:56
405 查看
问题描述
操作给定的二叉树,将其变换为源二叉树的镜像。思路描述
代码比文字更直观文字描述:新建一个二叉树,利用递归法,将源二叉树上的左节点赋值到新二叉树的右节点,将源二叉树上的右节点赋值到新二叉树的左节点。
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)
相关文章推荐
- 剑指offer之二叉树的镜像(Python)
- 面试题19_二叉树的镜像——剑指offer系列
- 剑指offer:重建二叉树(Python)
- 《剑指offer》——二叉树的镜像
- 《剑指offer》:[19]二叉树的镜像
- 剑指Offer:面试题19——二叉树的镜像(java实现)
- 《剑指Offer》之“二叉树的镜像”
- 剑指offer第19题 输出二叉树的镜像 java解法
- 剑指Offer(18)______二叉树的镜像
- 剑指Offer之二叉树的镜像
- 剑指offer(十八)二叉树的镜像
- python剑指offer系列层序遍历二叉树
- 剑指offer-面试题.二叉树的镜像
- 剑指offer 面试题19—二叉树的镜像
- 剑指Offer----二叉树的镜像
- 剑指Offer——二叉树的镜像
- 剑指offer系列之17:二叉树的镜像
- python剑指offer系列按之字形顺序打印二叉树
- 剑指offer-二叉树的镜像
- 《剑指Offer》学习笔记--面试题19:二叉树的镜像