二叉搜索树与双向链表(java)

2018-06-08  本文已影响0人  夏臻Rock

题目描述:

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

分析:

二叉搜索树转换为排序双向链表

由于要求转换后的链表有序,很自然的想到中序遍历二叉树。
当遍历到根节点的时候,我们把树分成三部分:值为10的结点,根节点为6的左子树、根节点为14的右子树。根据排序链表的定义,值为10的结点将和它的左子树的最大的一个结点(即值为8的结点)链接起来,同时它还将和右子树最小的结点(即值为12的结点)链接起来,如图:


把二叉搜索树分成三部分

显然,对于左右子树如何转换成一个排序的链表,其排序和转换的过程是一样的,自然要用到递归。

解答:

/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

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

    }

}
*/
public class Solution {
    private TreeNode pre = null; //保存当前节点的前一个节点
    private TreeNode head = null;//保存链表的头结点
    public TreeNode Convert(TreeNode pRootOfTree) {
        if(pRootOfTree==null) return null;
        inOrder(pRootOfTree);
        return head;
    }
    private void inOrder(TreeNode node) {
        if (node == null) return;
        inOrder(node.left);
        node.left = pre;
        if (pre != null)  pre.right = node;
        pre = node;
        if (head == null) head = node;
        inOrder(node.right);
    }
    
}
运行结果

方法二:

如上是递归调用中序遍历的方法,下面给出不使用递归方法的解法,其实更加适合理解。在非递归方法中,我们用一个栈来保存找到树的最左节点时走过的了路径,然后依次出栈,修改出栈元素的left和right指针即可。

/*如下,是非递归的方法:用一个栈来保存遍历的顺序*/
    public static TreeNode ConvertWithStack(TreeNode pRootOfTree) {
        //如果根节点为null,则直接返回null
        if (pRootOfTree == null) {
            return null;
        }
        TreeNode newHead = null;
        //否则,用cur记录当前节点,初始时指向根节点
        TreeNode cur = pRootOfTree;
        //prenode用来记录当前节点的前一个节点
        TreeNode prenode = null;
        Stack<TreeNode> stack = new Stack<TreeNode>();
        while (!stack.isEmpty() || cur != null) {
            //左子树上的节点入栈
            while (cur != null) {
                stack.push(cur);
                cur = cur.left;
            }
            //到了树的最左边时,将栈中的节点出栈
            cur = stack.pop();
            //初始时,prenode为null,将最左边的节点赋值给newHead,也同时赋值给prenode
            if (prenode == null) {
                newHead = cur;
                prenode = cur;
            }
            //prenode有值的时候,让前一个指针的右节点指向当前节点,当前节点的左指针指向前一个节点
            else {
                prenode.right = cur;
                cur.left = prenode;
                prenode = cur;
            }
            //遍历当前节点的右孩子
            cur = cur.right;
        }
        return newHead;
    }
上一篇 下一篇

猜你喜欢

热点阅读