深度优先搜索DFS | Morris遍历:力扣99. 恢复二叉搜索树

最新推荐文章于 2022-06-12 23:01:42 发布

剑圣土豆

最新推荐文章于 2022-06-12 23:01:42 发布

阅读量188

点赞数

分类专栏： LeetCode高频面试题

本文链接：https://blog.csdn.net/weixin_42042056/article/details/107887408

版权

LeetCode高频面试题专栏收录该内容

166 篇文章 5 订阅

订阅专栏

1、题目描述：

在这里插入图片描述

2、题解：

方法1：中序遍历迭代
二叉搜索树的性质：
1、若任意节点的左子树不空，则左子树上所有节点的值均小于它的根节点的值；
2、若任意节点的右子树不空，则右子树上所有节点的值均大于它的根节点的值；
3、任意节点的左、右子树也分别为二叉查找树；
4、没有键值相等的节点。
思路：

二叉搜索树的中序遍历是有序的
比如[4,2,3,1] 交换 4和1即可
代码精髓：
设置两个节点：
    第一个节点，是第一个按照中序遍历时前一个节点大于后一个节点，的前一个节点，即4
    第二个节点，是第一个节点找到后，再出现前一个节点大于后一个节点的，后一个节点，即1
    if not self.firstNode and self.pre.val >= root.val:
        self.firstNode = self.pre
    if self.firstNode and self.pre.val >= root.val:
    	self.secondNode = root
找到这两个节点后，交换节点值即可

Python代码如下：

# 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.
        """
        #中序遍历迭代
        firstNode = None
        secondNode = 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

方法2：中序遍历递归

思路：

二叉搜索树的中序遍历是有序的
比如[4,2,3,1] 交换 4和1即可
代码精髓：
设置两个节点：
    第一个节点，是第一个按照中序遍历时前一个节点大于后一个节点，的前一个节点，即4
    第二个节点，是第一个节点找到后，再出现前一个节点大于后一个节点的，后一个节点，即1
找到这两个节点，交换值即可

Python代码如下：

# 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.
        """
        # 中序遍历递归
        self.firstNode ,self.secondNode = None,None
        self.pre = TreeNode(float('-inf'))
        def in_order(root):
            if not root:return
            in_order(root.left)
            if not self.firstNode and self.pre.val >= root.val:
                self.firstNode = self.pre
            if self.firstNode and self.pre.val >= root.val:
                self.secondNode = root
            self.pre = root
            in_order(root.right)
        in_order(root)
        self.firstNode.val,self.secondNode.val = self.secondNode.val,self.firstNode.val

方法3：Morris中序遍历

可以参考力扣94题
面试算法：二叉树的Morris遍历算法

# 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.
        """
        # Morris中序遍历
        firstNode,secondNode = None,None
        pre = TreeNode(float('-inf'))
        cur = root
        while cur:
            if not cur.left:
                if not firstNode and pre.val >= cur.val:
                    firstNode = pre
                if firstNode and pre.val >= cur.val:
                    secondNode = cur
                pre = cur
                cur = cur.right
            else:
                #找左子树的最右边的节点
                temp = cur.left
                while temp.right and temp.right != cur:
                    temp = temp.right
                if not temp.right:
                    temp.right = cur
                    cur = cur.left
                if temp.right == cur:
                    temp.right = None
                    if not firstNode and pre.val >= cur.val:
                        firstNode = pre
                    if firstNode and pre.val >= cur.val:
                        secondNode = cur
                    pre = cur
                    cur = cur.right
        firstNode.val ,secondNode.val = secondNode.val,firstNode.val