class Solution(object):
# 中序遍历会发现一个或两个降序点
# 示例: [2,4,3,5](一个降序点); [5,3,4,2](两个降序点)
# 把降序点找到然后把他们的值交换
# 用全局变量保存前一个结点,第一个要交换的点和第二个要交换的点,以及记录第几次发现降序点
# 如果是一个降序点的话,要交换的值是发现降序点位置的前一个和后一个
# 如果是两个降序点的话,交换的值是第一次降序点的前一个和第二次降序点的后一个
def __init__(self):
self.firstNode = None
self.secondNode = None
self.pre = None
self.flag = True
def recoverTree(self, root):
"""
:type root: TreeNode
:rtype: None Do not return anything, modify root in-place instead.
"""
if not root:
return root
if not root.left and not root.right:
return root
self.inOrder(root)
self.firstNode.val, self.secondNode.val = self.secondNode.val, self.firstNode.val
def inOrder(self, root):
if not root:
return
# 中序遍历
self.inOrder(root.left)
if self.pre and self.pre.val > root.val:
if self.flag:
self.firstNode = self.pre
self.secondNode = root
self.flag = False
else:
self.secondNode = root
# 当前步执行完之后设置pre = root
self.pre = root
self.inOrder(root.right)
99. 存在两个错误交换结点的二叉搜索树
猜你喜欢
转载自blog.csdn.net/weixin_34258078/article/details/90926403
今日推荐
周排行