python第22天

最新推荐文章于 2025-03-27 16:42:40 发布

原创最新推荐文章于 2025-03-27 16:42:40 发布 · 166 阅读

0 ·

CC 4.0 BY-SA版权

本文探讨了在不改变二叉搜索树结构的情况下，如何恢复因两个节点错误交换而导致的树结构。通过迭代思想，深入理解如何定位并纠正错误节点，解决平行及相邻节点调换问题。

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

请在不改变其结构的情况下，恢复这棵树。

示例 1:

输入: [1,3,null,null,2]

1
/
3

2

输出: [3,1,null,null,2]

3
/
1

2
示例 2:

输入: [3,1,4,null,null,2]

3
/
1 4
/
2

输出: [2,1,4,null,null,3]

2
/
1 4
/
3`

关于这里我学习的是迭代的思想，一直往左走走到不能再走是然后入栈，排序比较大小找到被交换的两个数，这里有几个注意的点，一个是如果相邻的两个节点调换排序的问题。一个是关于平行调换的问题。

`python
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