剑指offer-二叉搜索树与双向链表

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

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

剑指offer-二叉搜索树与双向链表_第1张图片

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

由于要求转换之后的链表是排好序的,我们可疑中序遍历树中的每一个结点,这是因为中序遍历算法的特点是按照从小到达的顺序遍历二叉树的每一个结点。当遍历到根节点的时候,我们把树堪称三部分:值为10的结点,根节点为6的左子树、根节点为14的右子树。根据排序链表的定义,值为10的结点将和它的左子树的最大的一个结点(即值为8的结点)链接起来,同时它还将和右子树最小的结点(即值为12的结点)链接起来,如图:

剑指offer-二叉搜索树与双向链表_第2张图片

按照中序遍历的顺序,当我们遍历转换到根节点(值为10的结点)时,它的左子树已经转换成一个排序的链表了,并且处在链表中的最后一个结点是当前值的最大的结点。我们把值为8的结点根节点链接起来,此时链表中的最后一个结点是10了。接着我们去遍历转换右子树,并把根节点和右子树最小的结点链接起来。至于怎么去转换它的左子树和右子树,由于遍历和转换过程是一样的,我们自然的想到了递归。

Java代码实现:(牛客网未通过的

public class Solution {
	public TreeNode Convert(TreeNode pRootOfTree) {
		TreeNode pLastNode = null;
		ConvertNode(pRootOfTree, pLastNode);

		TreeNode pHead = pLastNode;
		while (pHead != null && pHead.left != null) {
			pHead = pHead.left;
		}
		return pHead;
	}

	private void ConvertNode(TreeNode pRootOfTree, TreeNode pLastNode) {
		// TODO Auto-generated method stub
		if(pRootOfTree==null){
			return;
		}
		TreeNode pCurrent=pRootOfTree;
		if(pCurrent.left!=null){
			ConvertNode(pCurrent.left, pLastNode);
		}
		pCurrent.left=pLastNode;
		if(pLastNode!=null){
			pLastNode.right=pCurrent;
		}
		pLastNode=pCurrent;
		if(pCurrent.right!=null){
			ConvertNode(pCurrent.right, pLastNode);
		}
	}
}

class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

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

    }
}

通过的(转载):

public class Solution {
    public TreeNode Convert(TreeNode root) {
 if(root==null)
            return null;
        if(root.left==null&&root.right==null)
            return root;
        // 1.将左子树构造成双链表,并返回链表头节点
        TreeNode left = Convert(root.left);
        TreeNode p = left;
        // 2.定位至左子树双链表最后一个节点
        while(p!=null&&p.right!=null){
            p = p.right;
        }
        // 3.如果左子树链表不为空的话,将当前root追加到左子树链表
        if(left!=null){
            p.right = root;
            root.left = p;
        }
        // 4.将右子树构造成双链表,并返回链表头节点
        TreeNode right = Convert(root.right);
        // 5.如果右子树链表不为空的话,将该链表追加到root节点之后
        if(right!=null){
            right.left = root;
            root.right = right;
        }
        return left!=null?left:root;
    }
}
添加笔记


你可能感兴趣的:(算法,递归,链表,遍历,搜索)