Recover Binary Search Tree@LeetCode

最新推荐文章于 2025-08-12 18:32:44 发布

weixin_33757911

最新推荐文章于 2025-08-12 18:32:44 发布

阅读量59

点赞数

CC 4.0 BY-SA版权

文章标签：数据结构与算法

原文链接：https://segmentfault.com/a/1190000002688926

本文介绍了一种通过中序遍历检测并修复被错误交换节点的二叉搜索树的方法。利用BST特性，当发现两个节点顺序错误时，记录这两个节点，并进行值交换，从而恢复树的正确结构。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Recover Binary Search Tree

根据BST树的特性来，对BST的中序遍历，得到的是一个升序数列。所以在遍历过程中检测出两个异常的位置，对其进行交换即可。

一旦有两个位置的节点被交换了，那么中序遍历就会出现有两个：Node[i] > Node[i + 1]其中i是错误位置，Node[j] < Node[j - 1]其中j是错误位置，遵循这个规律，找到相应的Node[i]和Node[j]对其进行交换（只交换val值）即可。

实现代码如下：

javapublic class Solution {

    private TreeNode wrongLessNode;
    private TreeNode wrongLargerNode;
    private TreeNode preNode;

    public void recoverTree(TreeNode root) {
        recover(root);
        if (wrongLessNode != null && wrongLargerNode != null) {
            int temp = wrongLessNode.val;
            wrongLessNode.val = wrongLargerNode.val;
            wrongLargerNode.val = temp;
        }
    }

    private void recover(TreeNode root) {
        if (root == null)
            return;
        if (preNode == null && root.left == null) {
            preNode = root;
        }
        recover(root.left);
        if (preNode != null && root.val < preNode.val) {
            if (wrongLessNode == null) {
                wrongLessNode = preNode;
                wrongLargerNode = root;
            }
            else {
                wrongLargerNode = root;
                return;
            }
        }
        preNode = root;
        recover(root.right);
    }
}