- Recover Binary Search Tree
题目描述:
Two elements of a binary search tree (BST) are swapped by mistake.
Recover the tree without changing its structure.
解决思路:
题目要求把一个有序二叉树恢复,其中的两个值调换了,要修改过来。
题目给出的list是广度遍历出来的list,所以看不出什么。最先想到的是用inorder(中序遍历)来做,因为inorder遍历出来的list是有序的列表,会看得清晰一些。
举个例子来分析几种情况:
这是一颗正常的二叉树,中序遍历出来是有序的。
如果是有问题的树,那就可能有这几种情况。很明显,树1~3中序遍历出来一看就不是有序的(不是从小到大排列)。
所以我觉得要做的:中序遍历的时候,将当前值(root.val)和pre(前一个值)比较,当前值肯定要比pre小,不然就有问题,就得记录下来(放在vec =[ ] 里面)。
?:那么要怎么记录,记录哪一个?
!:看树2,当两个数连在一起的时候,就两个都得记,否则就会漏掉。树1和树2会怎么样,看下图
也就是说,如果一发现问题就把pre和当前值都记下来。vec放的东西,我们只要关注头和尾就行,也就是vec[0]和vec[-1]。
最后就是修复的问题,思路和上面一样,找一个变量pre_root记录前一个子树,queue =[ ]来放有问题的子树(作用是跟vec一样的)。
(我的理解,中序遍历就是把每一个节点都当成一个子树,按照最左下的子树开始,左中右一直顺序执行,对子树的root.val进行操作)
变量菜单:
self.vec = []
self.pre = float("-inf")
self.queue = []
self.pre_root=None
(浪费了一点存储空间,当然可以用更少变量,为了思路清晰就不优化了)
我的python代码:
(不会传代码,可能缩进有点问题)
def __init__(self):
self.vec = []
self.pre = float("-inf")
self.queue = []
self.pre_root=None
def recoverTree(self, root):
"""
:type root: TreeNode
:rtype: void Do not return anything, modify root in-place instead.
"""
if root == None: return
self.inorder(root)
#开始修复
self.queue[0].val=self.vec[-1]
self.queue[-1].val=self.vec[0]
#修复结束
def inorder(self, root):
if root == None: return
if root.val == None:return
self.inorder(root.left)
#开始处理
if root.val < self.pre:
self.queue.append(self.pre_root)
self.queue.append(root)
self.vec.append(self.pre)
self.vec.append(root.val)
self.pre = root.val #记录值
self.pre_root = root #记录树
#处理结束
self.inorder(root.right)