【8月打卡~Leetcode每日一题】99. 恢复二叉搜索树(难度:困难)

99. 恢复二叉搜索树
二叉搜索树中的两个节点被错误地交换。

请在不改变其结构的情况下,恢复这棵树
【8月打卡~Leetcode每日一题】99. 恢复二叉搜索树(难度:困难)_第1张图片
二叉搜索树的性质:左子树节点值 < 根节点值 < 右子树节点值
由此很自然的想到中序遍历,如果这个二叉搜索树是正常的,则中序遍历节点值的结果应该是个升序数组,考虑其中两个节点互换,会有两种情况①互换相邻节点,则数组中有一个数不满足升序②互换不相邻节点,则数组中有两个数不满足升序

# 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为节点数
【8月打卡~Leetcode每日一题】99. 恢复二叉搜索树(难度:困难)_第2张图片

你可能感兴趣的:(python,数据结构,算法,二叉树)