剑指offer(牛客)---26.二叉搜索树与双向链表

题目描述

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

 

牛客推荐:

/**
public 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 pRootOfTree) {
        if(pRootOfTree==null)
            return null;
        if(pRootOfTree.left==null&&pRootOfTree.right==null)
            return pRootOfTree;
        // 1.将左子树构造成双链表，并返回链表头节点
        TreeNode left = Convert(pRootOfTree.left);
        TreeNode p = left;
        // 2.定位至左子树双链表最后一个节点
        while(p!=null&&p.right!=null){
            p = p.right;
        }
        // 3.如果左子树链表不为空的话，将当前root追加到左子树链表
        if(left!=null){
            p.right = pRootOfTree;
            pRootOfTree.left = p;
        }
        // 4.将右子树构造成双链表，并返回链表头节点
        TreeNode right = Convert(pRootOfTree.right);
        // 5.如果右子树链表不为空的话，将该链表追加到root节点之后
        if(right!=null){
            right.left = pRootOfTree;
            pRootOfTree.right = right;
        }
        return left!=null?left:pRootOfTree;       
    }
}

 

解题思路:

假设有如下搜索二叉树:

 第一次判断一直到 TreeNode left = Convert(pRootOfTree.left);,然后一直遍历到结点6之后判断它的左右孩子都为null,返回null,现在left=null;pRootOfTree=节点6;

然后一直判断都为null,所以递归pRootOfTree(节点6)的右孩子( TreeNode right = Convert(pRootOfTree.right)),右孩子为空,返回null,right=null,最后判断返回当前pRootOfTree(6);然后继续递归,回到pRootOfTree(节点9),返回left=6,p=left=6然后一直判断到 if(left!=null),进入将节点9接到结点6的右孩子上面,结点6放到节点9的最还在上面;

相当于:

 又递归TreeNode right = Convert(pRootOfTree.right),pRootOfTree(9),有右孩子,右孩子10没有左右还在,所以直接返回当前结点10,将节点10的左孩子指向pRootOfTree(结点9),结点9的右孩子指向10,结构如图所示:

又递归完毕,往上走,TreeNode left = Convert(pRootOfTree.left);到了pRootOfTree(节点12),left(节点9),p=left(节点9),

while(p!=null&&p.right!=null){
            p = p.right;
        }

满足条件:这时候p的右孩子是10,所以当前p(结点10);

if(left!=null){
            p.right = pRootOfTree;
            pRootOfTree.left = p;
        }

满足条件:pRootOfTree(结点12)加入到排序链表中将当前结点如下图所示:

 后面就不在分析了,一直递归判断就可以了;

总结:从最左子树开始判断,因为他是链表的头,然后如果有右孩子就放他后面,没有的递归回去,父节点放到它后面,然后判断父节点有没有右孩子,有的话放到后面接起来,然后继续递归,巧妙之处在于right作为链表的下一个指向结点,left作为父节点.