剑指offer第二版面试题36:二叉搜索树与双向链表(java)

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

比如如下图中的二叉搜索树,则输出转换之后的排序双向链表为:
这里写图片描述

分析:

在二叉树中,每个结点都有两个指向子节点的指针。在双向链表中,每个结点也有两个指针,他们分别指向前一个结点和后一个结点。由于这两种结点的结构相似,同时二叉搜索树也是一种排序的数据结构,因此在理论上有可能实现二叉搜索树和排序的双向链表的转换。在搜索二叉树中,左子结点的值总是小于父节点的值,右子节点的值总是大于父节点的值。因此我们在转换成排序的双向链表时,原先指向的左子结点的指针调整为链表中指向前一个结点的指针,原先指向右子节点的指针调整为链表中指向后一个结点的指针。接下来我们考虑如何转化。

由于要求转换之后的链表是排好序的,我们可以中序遍历树中的每一个结点,这是因为中序遍历算法的特点是按照从小到大的顺序遍历二叉树的每一个结点。当遍历到根节点的时候,我们把树分为三部分:值为10的结点,根节点为6的左子树、根节点为14的右子树。根据排序链表的定义,值为10的结点将和它的左子树的最大的一个结点(即值为8的结点)连接起来,同时它还将右子树最小的结点(即值为12的结点)连接起来,如图:
这里写图片描述
按照中序遍历的顺序,当我们遍历转换到根节点(值为10的结点)时,它的左子树已经转换成一个排序的链表了,并且处在链表中的最后一个结点是当前值的最大的结点。我们把值为8的结点与根节点链接起来,此时链表中的最后一个结点是10了。接着我们去遍历转换右子树,并把根节点和右子树最小的结点链接起来。至于怎么去转换它的左子树和右子树,由于遍历和转换过程是一样的,我们自然的想到了递归。

代码如下:

/**
 * 二叉搜索树转换成排序的双向链表
 */
public class BinaryTreeConvert {

    public BinaryTreeNode convert(BinaryTreeNode root) {
        //指向双向链表的尾节点
        BinaryTreeNode lastNodeInList = null;
        convertNode(root, lastNodeInList);

        //我们需要返回头结点
        BinaryTreeNode head = lastNodeInList;
        while(head != null && head.left != null){
            head = head.left;
        }
        return head;
    }

    //使用递归进行转换
    public void convertNode(BinaryTreeNode node, BinaryTreeNode lastNodeInList) {
        if(node == null)
            return ;

        BinaryTreeNode currentNode = node;
        if(currentNode.left != null){
            convertNode(currentNode.left, lastNodeInList);
        }
        currentNode.left = lastNodeInList;
        if(lastNodeInList != null){
            lastNodeInList.right = currentNode;
        }
        lastNodeInList = currentNode;
        if(currentNode.right != null){
            convertNode(currentNode.right, lastNodeInList);
        }
    }

}

//二叉树节点
class BinaryTreeNode {
    int value;
    BinaryTreeNode left;
    BinaryTreeNode right;
}

猜你喜欢

转载自blog.csdn.net/weixin_37672169/article/details/80563026