【LeetCode】99. Recover Binary Search Tree 解题报告(Python)

127 篇文章 2 订阅
102 篇文章 2 订阅

题目分析:

二叉搜索树中的两个节点被错误地交换。请在不改变其结构的情况下,恢复这棵树。
示例 1:
在这里插入图片描述

解题思路:

第一种:中序遍历将数值取出来,然后排序,然后再插入进入,使用的空间复杂度是O(n),见提交代码1。
第二种:使用指针找到变动的那两个值,空间复杂度为常量见提交代码2.

  1. 设置n1,n2的值用于寻找那两个点,self.n1 = TreeNode(None); self.n2 = TreeNode(None)
  2. 设置pre节点用于保存上一个节点,根据上一个节点判断当前节点是否是错误的self.pre = TreeNode(None)
  3. 中序遍历找到root.val < self.pre.val保存为n1,n2
  4. 交换n1,n2的值

第二种过程如下图:
在这里插入图片描述
很明显3和5颠倒了。那么在中序遍历时:当碰到第一个逆序时:为5->4,那么将n1指向5,n2指向4,注意,此时n1已经确定下来了。然后pre和root一直向后遍历,直到碰到第二个逆序时:4->3,此时将n2指向3,那么n1和n2都已经确定,只需要交换节点的值即可。pre指针用来比较中序遍历中相邻两个值的大小关系。

提交代码1:(递归+列表,Runtime: 80 ms, faster than 95.07% )

class Solution:
    def inorder(self,  root: TreeNode, list_num: list, list_root: list) -> None:
        if root:
            self.inorder(root.left, list_num, list_root)
            list_num.append(root.val);list_root.append(root)
            self.inorder(root.right, list_num, list_root)
    def recoverTree(self, root: TreeNode) -> None:
        list_nums = []
        list_root = []
        self.inorder(root, list_nums, list_root)
        list_nums.sort()
        for i, root in enumerate(list_root):
            root.val = list_nums[i]

提交代码2:(递归+指针,Runtime: 80 ms, faster than 95.07% )

class Solution:

    def inorder(self,  root: TreeNode) -> None:
        if root:
            self.inorder(root.left)
            if self.pre and root.val != None and self.pre.val != None and root.val < self.pre.val:
                self.n2 = root
                if self.n1.val == None: self.n1 = self.pre
            self.pre = root
            self.inorder(root.right)

    def recoverTree(self, root: TreeNode) -> None:
        self.n1 = TreeNode(None); self.n2 = TreeNode(None)
        self.pre = TreeNode(None)
        self.inorder(root)
        self.n1.val, self.n2.val = self.n2.val, self.n1.val

参考博客

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值