Java二叉搜索树转换成一个排序的双向链表

最新推荐文章于 2022-08-27 09:52:36 发布

这瓜保熟么

最新推荐文章于 2022-08-27 09:52:36 发布

阅读量862

点赞数

分类专栏：简单小算法

原文链接：https://zhuanlan.zhihu.com/p/113139103

版权

简单小算法专栏收录该内容

143 篇文章 2 订阅

订阅专栏

题目描述：输入一棵二叉搜索树，将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点，只能调整树中结点指针的指向。比如输入下图中二叉搜索树，则输出转换后的排序双向链表

在二叉搜索树中，每个结点都有两个分别指向其左、右子树的指针，左子树结点的值总是小于父结点的值，右子树结点的值总是大于父结点的值。在双向链表中，每个结点也有两个指针，它们分别指向前一个结点和后一个结点。所以这两种数据结构的结点是一致，二叉搜索树和双向链表，只是因为两个指针的指向不同而已，通过改变其指针的指向来实现是完全可能的。举个例子，如下图中的二叉搜索树，转换后的双向链表为：

为了减少指针的变换次数，并让操作更加简单，在转换成排序双向链表时，原先指向左子结点的指针调整为链表中指向前一个结点的指针，原先指向右子结点的指针调整为链表中指向下一个结点的指针。由于要求链表是有序的，可以借助二叉树中序遍历，因为中序遍历算法的特点就是从小到大访问结点。当遍历访问到根结点时，假设根结点的左侧已经处理好，只需将根结点与上次访问的最近结点（左子树中最大值结点）的指针连接好即可。进而更新当前链表的最后一个结点指针。同时中序遍历过程正好是转换成链表的过程，可采用递归方法处理。

package cn.dzp.flyroc.offer;

import java.util.Stack;

public class TreeConvertListDemo {

    /*
    * 题目描述：输入一颗二叉搜索树，将该二叉搜索树转换成一个排序的双向链表。
    * 要求不能创建任何新的节点，只能调整树中节点指针的指向   * */


    /*思路：
    * ，在转换成排序双向链表时，原先指向左子结点的指针调整为链表中指向前一个结点的指针，
    * 原先指向右子结点的指针调整为链表中指向下一个结点的指针。
      由于要求链表是有序的，可以借助二叉树中序遍历，
      因为中序遍历算法的特点就是从小到大访问结点。
    * */



    //代码实现

    //定义TreeNode类
    public static class TreeNode {

        int val;
        TreeNode left = null;
        TreeNode right = null;

        TreeNode(int val) {
            this.val = val;
        }
    }

    /*
    * 递归方法
    * */
    public class Convert {

        // 双向链表的左边头结点和右边头节点
        TreeNode leftHead = null;
        TreeNode rightHead = null;

        public TreeNode treeConvertListOfCircle(TreeNode root){

            // 递归调用叶子节点的左右节点返回null
            if (root == null)
                return null;

            // 第一次运行时，它会使最左边叶子节点为链表的第一个节点
            treeConvertListOfCircle(root.left);

            if (rightHead == null) {

                leftHead = root;
                rightHead = root;

            } else {

                // 把根节点插入到双向链表右边，rightHead向后移动
                rightHead.right = root;
                root.left = rightHead;
                rightHead = root;

            }

            // 把右叶子节点也插入到双向链表（rightHead已确定，直接插入）
            treeConvertListOfCircle(root.right);

            // 返回左边头结点
            return leftHead;
        }
    }