[Leetcode 每日一题] 恢复二叉搜索树

最新推荐文章于 2024-09-13 18:30:00 发布

诗9趁年华

最新推荐文章于 2024-09-13 18:30:00 发布

阅读量140

点赞数

分类专栏：每日一题文章标签：数据结构算法 leetcode java 二叉树

本文链接：https://blog.csdn.net/qq_36565596/article/details/107875573

版权

每日一题专栏收录该内容

8 篇文章 0 订阅

订阅专栏

Level：Hard

题目：

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

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

示例 :
在这里插入图片描述

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

思路：

要充分利用二叉搜索树的特性。二叉搜索树的中序遍历得到的序列是递增。假设递增序列 $a = [1, 2, 3, 4, 5, 6, 7]$ 变成了 $a = [1, 6, 3, 4, 5, 2, 7]$ ，显然可以发现有两个位置不满足 $a_i < a_{i+1}$ 。因此本题的任务就是需要找到这两个位置并交换即可。

中序遍历二叉树搜索树
找到不满足条件的位置
遍历后的递增数列中，如果是相邻的两个数交换，那么仅有一个位置不满足条件，否则有两个。如果有一个，标记为 $i$ ,那么对应被错误交换的结点是 $a_i$ 和 $a_{i+1}$ 。
如果是有两个结点不满足条件，标记为 $i$ 和 $j$ ( $i < j$ 且 $a_i > a_{i+1}$ && $a_j > a_{j+1}$ )，那么对应被错误交换的结点是 $a_i$ 和 $a_{j+1}$ 。
交换找到的两个待交换的结点即可

开辟一个新数组 $n u m s$ 来记录中序遍历得到的值序列，然后线性遍历找到两个位置 $i$ 和 $j$ ，并重新遍历原二叉搜索树修改对应节点的值完成修复，具体实现可以看下面的代码。

代码：

/**
 * Definition for a binary tree node.
 * public 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;
 *     }
 * }
 */
class Solution {
    public void recoverTree(TreeNode root) {
        List<Integer> nums = new ArrayList<Integer>();
        inorder(root, nums);
        int[] swapped = findTwoSwapped(nums); // 得到待交换的两个数
        recover(root, 2, swapped[0], swapped[1]);
    }

	// 递归中序遍历二叉搜索树
    public void inorder(TreeNode root, List<Integer> nums){
        if(root == null)
            return;
        inorder(root.left, nums); // 左
        nums.add(root.val); // 根
        inorder(root.right, nums); // 右
    }

    public int[] findTwoSwapped(List<Integer> nums){
        int n = nums.size();
        int x = -1, y = -1;
        for (int i = 0; i < n -1; ++i){
            if(nums.get(i+1) < nums.get(i)){ // 后一个数比前一个数小，
                y = nums.get(i+1);
                if(x == -1){
                    x = nums.get(i);
                }else{
                    break;
                }

            }
        }
        return new int[]{x, y};
    }

    public void recover(TreeNode root, int count, int x, int y){
        if (root != null){
            if(root.val == x || root.val == y){
                root.val = root.val == x ? y : x;
                if(--count == 0)
                return;
            }
            recover(root.right, count, x, y);
            recover(root.left, count, x, y);
        }
    }
}

复杂度分析

时间复杂度： $O (N)$ ，其中 $N$ 为二叉搜索树的节点数。中序遍历需要 $O (N)$ 的时间，判断两个交换节点在最好的情况下是 $O (1)$ ，在最坏的情况下是 $O (N)$ ，因此总时间复杂度为 $O (N)$ 。
空间复杂度： $O (N)$ 。我们需要用 $n u m s$ 数组存储树的中序遍历列表。