LeetCode刷题笔录Recover Binary Search Tree

最新推荐文章于 2024-09-15 15:43:59 发布

likecool21

最新推荐文章于 2024-09-15 15:43:59 发布

阅读量607

点赞数

分类专栏： LeetCode 文章标签： leetcode traversal

本文链接：https://blog.csdn.net/likecool21/article/details/39801041

版权

LeetCode 专栏收录该内容

114 篇文章 0 订阅

订阅专栏

这篇博客探讨了如何在不改变结构的情况下恢复二叉搜索树，通过中序遍历来检测并解决节点错误交换的问题。对于两种情况——相邻元素和非相邻元素的交换，作者提供了识别和交换正确节点的方法。

摘要由CSDN通过智能技术生成

Two elements of a binary search tree (BST) are swapped by mistake.

Recover the tree without changing its structure.

Note:
A solution using O( n ) space is pretty straight forward. Could you devise a constant space solution?

confused what "{1,#,2,3}" means? > read more on how binary tree is serialized on OJ.

和Valid Binary Search Tree那题比较像，也是要做一个in order traversal，过程中比较当前node的值和之前一个node的值得大小，看看有没有反序的情况。

有两种情况：1是两个相邻的元素被交换了，比如12345678 -> 12354678，反序只会发生一次，因此记录5，4两个node就对了

2是不相邻元素被交换了，比如12345678 -> 12745638,反序会发生两次，一次在74一次在63.可以看出需要交换的元素是7和3，即第一次反序的前一个元素和第二次反序的后一个元素。

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public void recoverTree(TreeNode root) {
        ArrayList<TreeNode> pre = new ArrayList<TreeNode>();
        pre.add(null);
        ArrayList<TreeNode> nodes = new ArrayList<TreeNode>();
        inorder(root, pre, nodes);
        
        int temp = nodes.get(0).val;
        nodes.get(0).val = nodes.get(1).val;
        nodes.get(1).val = temp;
    }
    
    public void inorder(TreeNode root, ArrayList<TreeNode> pre, ArrayList<TreeNode> nodes){
        if(root == null)
            return;
        
        inorder(root.left, pre, nodes);
        
        if(pre.get(0) != null && root.val < pre.get(0).val){
            if(nodes.size() == 0){
                nodes.add(pre.get(0));
                nodes.add(root);
            }
            else{
                nodes.set(1, root);
            }
        }

        pre.set(0, root);
        inorder(root.right, pre, nodes);
        
    }
}