leetcode99. 恢复二叉搜索树

最新推荐文章于 2025-10-29 13:09:05 发布

原创最新推荐文章于 2025-10-29 13:09:05 发布 · 90 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

leetcode-python 专栏收录该内容

150 篇文章

订阅专栏

本文介绍了一种在不改变二叉搜索树结构的情况下，通过中序遍历找到并修复两个错位节点的方法。利用节点值大小关系，将错位节点存储并最终交换其值，实现树的恢复。

题目：

二叉搜索树中的两个节点被错误地交换。
请在不改变其结构的情况下，恢复这棵树。

示例 1:
在这里插入图片描述
示例 2:

思路：
中序遍历这颗二叉搜索树，根据节点值的大小可以很容易的找到错位的节点，把错位的节点存起来，最后交换错位节点的值。

代码：

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution(object):
    def recoverTree(self, root):
        """
        :type root: TreeNode
        :rtype: None Do not return anything, modify root in-place instead.
        """
        stack = []
        order = []
        res = []

        while root or stack:
            if root:
                stack.append(root)
                root = root.left
            else:
                temp = stack.pop()
                if order:
                    if order[-1].val > temp.val:
                        res.append(order[-1])
                        res.append(temp)
                order.append(temp)
                if temp.right:
                    root = temp.right

        res[0].val, res[-1].val = res[-1].val, res[0].val