99. Recover Binary Search Tree

最新推荐文章于 2025-08-10 08:58:43 发布

转载最新推荐文章于 2025-08-10 08:58:43 发布 · 53 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/skillking/p/9711950.html

文章标签：

#数据结构与算法

本文介绍了一种算法，用于解决二叉查找树中两个元素位置错误的问题。通过中序遍历，找出两个错位节点并进行交换，恢复树的正确结构。

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

一、题目

　　1、审题

　　2、分析

　　　　给出一个二叉查找树，其中有两个元素的位置弄错了，写算法将其恢复。

二、解答

　　1、思路：

　　　　方法一、

　　　　　　通过中序遍历可以确定一棵二叉查找树由小到大的顺序。

　　　　　　所以在此错位的查找树中查找到的节点中有 1 个比后续节点值大，最后 1 个比前面相邻节点值小。交换这两个结点值即可。

　　TreeNode firstElement = null;
    TreeNode secondElement = null;
    TreeNode prevElement = new TreeNode(Integer.MIN_VALUE);
    
    public void recoverTree(TreeNode root) {
        
        traverse(root);
        int tmp = firstElement.val;
        firstElement.val = secondElement.val;
        secondElement.val = tmp; 
        
        
    }
    
    private void traverse(TreeNode root) {
        
        if(root == null)
            return;
        
        traverse(root.left);
        
        ///
        if(firstElement == null && prevElement.val >= root.val)
            firstElement = prevElement;    // 记录第一个比后续元素大的那个结点
        
        if(firstElement != null && prevElement.val >= root.val)
            secondElement = root;        // 记录最后一个比前一个结点小的节点。
        
        prevElement = root;        // 要查找的节点指针往后移动
        ///
        
        traverse(root.right);
    }