LeetCode: 99. 恢复二叉搜索树
二叉搜索树中的两个节点被错误地交换。
请在不改变其结构的情况下,恢复这棵树。
示例 1:
输入: [1,3,null,null,2]
1
/
3
\
2
输出: [3,1,null,null,2]
3
/
1
\
2
示例 2:
输入: [3,1,4,null,null,2]
3
/ \
1 4
/
2
输出: [2,1,4,null,null,3]
2
/ \
1 4
/
3
进阶:
使用 O(n) 空间复杂度的解法很容易实现。
你能想出一个只使用常数空间的解决方案吗?
思路:
先序遍历二叉树、利用前、后指针,将不满足 左>根>右 的节点记录下来,进行调换
二叉树定义:
# Definition for a binary tree node.
class TreeNode:
def __init__(self, x):
self.val = x
self.left = None
self.right = None
实现代码,基于迭代式先序遍历:
class Solution:
def recoverTree(self, root: TreeNode) -> None:
firstnode,secondnode = None,None
pre = TreeNode(float('-inf'))
stack =[]
p = root
while p or stack:
while p:
stack.append(p)
p = p.left
p = stack.pop()
if not firstnode and pre.val>p.val:
firstnode = pre
if firstnode and pre.val>p.val:
secondnode = p
pre = p
p = p.right
firstnode.val,secondnode.val = secondnode.val,firstnode.val