Leetcode:NO.99 恢复二叉搜索树中序遍历

最新推荐文章于 2022-06-19 19:26:02 发布

泛泛之素

最新推荐文章于 2022-06-19 19:26:02 发布

阅读量338

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/tonydz0523/article/details/107874735

版权

算法专栏收录该内容

288 篇文章 2 订阅

订阅专栏

该博客主要介绍了如何解决LeetCode中的第99题——恢复二叉搜索树。作者指出，由于树的中序遍历应该是递增的，因此可以通过中序遍历来获取不正确排序的节点。通过对中序遍历结果排序，找到位置错误的节点并交换它们的值，即可恢复树的正确状态。进阶讨论中提到，只需找到一个节点大于其后继节点以及一个节点小于其前驱节点的情况，就能确定需要交换的节点。

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

题目

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

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

示例 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

进阶:

使用 O(n) 空间复杂度的解法很容易实现。
你能想出一个只使用常数空间的解决方案吗？

链接：https://leetcode-cn.com/problems/recover-binary-search-tree

解题记录

应为是二叉搜索树，中序便利是递增的
为了不改变树的结构，优先中序便利二叉树，获得中序遍历顺序的数据
由于有两个位置错了，那么得到的数据自然不是递增的
将数组sort，得到递增数据
通过中序遍历，比对数据不对的更换即可

/**
 * @author: ffzs
 * @Date: 2020/8/8 上午8:26
 */

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode() {}
    TreeNode(int val) { this.val = val; }
    TreeNode(int val, TreeNode left, TreeNode right) {
        this.val = val;
        this.left = left;
        this.right = right;
    }
}

public class Solution {

    List<Integer> values;
    int index = 0;
    public void recoverTree(TreeNode root) {
        values = new ArrayList<>();
        getValue(root);
        values.sort(Comparator.naturalOrder());
        changeValue(root);
    }

    private void getValue (TreeNode root) {
        if (root != null) {
            getValue(root.left);
            values.add(root.val);
            getValue(root.right);
        }
    }

    private void changeValue (TreeNode root) {
        if (root != null) {
            changeValue(root.left);
            if (root.val != values.get(index)){
                root.val = values.get(index);
            }
            index ++;
            changeValue(root.right);
        }
    }
}

在这里插入图片描述

进阶

题目说只有两个节点发生了变化
那么可知在中序遍历的情况下，有一个节点比后一个节点大，而且有一个节点比前一个节点小，因为就是这两个节点的位置发生了变化
那么我们只需要将这两个节点找到，并且更换他们的value即可达到目的

/**
 * @author: ffzs
 * @Date: 2020/8/8 上午8:59
 */
public class Solution2 {

    TreeNode wrong1 = null;
    TreeNode wrong2 = null;
    TreeNode preNode = null;
    public void recoverTree(TreeNode root) {
        sortTree(root);

        int tmp = wrong1.val;
        wrong1.val = wrong2.val;
        wrong2.val = tmp;
    }

    private void sortTree (TreeNode root) {
        if (root != null) {
            sortTree(root.left);

            if (preNode == null) preNode = root;
            if (wrong1 == null && root.val < preNode.val) wrong1 = preNode;
            if (wrong1 != null && root.val < preNode.val) wrong2 = root;
            preNode = root;

            sortTree(root.right);
        }
    }
}