微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

99-恢复二叉搜索树

二叉搜索树的概念:
根节点要小于右子树的节点,必须要大于左子树的节点。

在这里插入图片描述


根据二分搜索树的特点,其中序遍历是有序的。

在这里插入图片描述


所以先对其进行排序

//中序遍历二叉树
    public void dfs(TreeNode node, List<TreeNode> list){
        if (node.left == null){
            return;
        }
        dfs(node.left,list);
        list.add(node);
        dfs(node.right,list);
    }

然后判断可能错误的节点

public void recoverTree(TreeNode root) {
        List<TreeNode> list = new ArrayList<>();
        dfs(root,list);
        TreeNode x = null;
        TreeNode y = null;
        //打印遍历的结果,遍历可能出现出错的节点 x 和 y
        for (int i = 0; i < list.size() - 1; i++) {
            if (list.get(i).val > list.get(i + 1).val){
                y = list.get(i + 1);
                if (x == null){
                    x = list.get(i);
                }
            }
        }
        //当 x 和 y 不为空是交换节点两个的值,恢复二叉树
        if (x != null && y != null){
            int temp = x.val;
            x.val = y.val;
            y.val = temp;
        }
    }

    //中序遍历二叉树
    public void dfs(TreeNode node, List<TreeNode> list){
        if (node.left == null){
            return;
        }
        dfs(node.left,list);
        list.add(node);
        dfs(node.right,list);
    }

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐