99. 恢复二叉搜索树
二叉搜索树中的两个节点被错误地交换。
请在不改变其结构的情况下,恢复这棵树
二叉搜索树的性质:左子树节点值 < 根节点值 < 右子树节点值
由此很自然的想到中序遍历,如果这个二叉搜索树是正常的,则中序遍历节点值的结果应该是个升序数组,考虑其中两个节点互换,会有两种情况①互换相邻节点,则数组中有一个数不满足升序②互换不相邻节点,则数组中有两个数不满足升序
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def recoverTree(self, root: TreeNode) -> None:
"""
Do not return anything, modify root in-place instead.
"""
Node_list = []
def LDR(root):
if not root:
return
LDR(root.left)
Node_list.append(root)
LDR(root.right)
LDR(root)
x,y = None,None
now_node = Node_list [0]
for i in range(1,len(Node_list)):
if(now_node.val>Node_list[i].val):
y = Node_list[i]
if not x :
x = now_node
now_node = Node_list[i]
x.val,y.val = y.val,x.val
时间空间复杂度都是O(N) N为节点数