这个题和上一篇博客里面的题思路有相通的地方,题目已经明确只有一对错误。我们只要找到这两个地方,然后将其交换即可。
展示代码:
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def compare(self,root):
if not root:
return
self.compare(root.left)
if self.re!=None and self.re.val>root.val:
if self.l1==None:
self.l1=self.re
self.l2=root
self.re=root
self.compare(root.right)
def recoverTree(self, root):
"""
:type root: TreeNode
:rtype: None Do not return anything, modify root in-place instead.
"""
self.l1=None
self.l2=None
self.re=None
if not root:
return []
self.compare(root)
self.l1.val,self.l2.val=self.l2.val,self.l1.val
self.l1,self.l2用来记录错误节点处。
self.re是从最左边开始的。