【剑指 offer】(Java)二叉搜索树与双向链表

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

/**
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) {
        //使用中序遍历得到有序序列
        //把相邻元素相互指向，left指向前一个元素，right指向后一个元素
        
        //1.特殊情况 空树or一个节点
        if(pRootOfTree == null){
            return null;
        }
        if(pRootOfTree.left == null && pRootOfTree.right == null){
            return pRootOfTree;
        }
        
        //2.递归把左子树变成双向链表，返回链表的第一个节点
        TreeNode left = Convert(pRootOfTree.left);
        //3.左子树链表 + root
        
        TreeNode leftTail = left;
        while(leftTail != null && leftTail.right != null){
            leftTail = leftTail.right;
        }
        //todo left为空
        if(left != null){
            leftTail.right = pRootOfTree;
            pRootOfTree.left = leftTail;
        }
        //4.递归把右子树变成双向链表，左子树链表 + root + 右子树链表
        TreeNode right = Convert(pRootOfTree.right);
        
        if(right != null){
            pRootOfTree.right = right;
            right.left = pRootOfTree;
        }
        
        //返回整个链表的头节点
        return left == null ? pRootOfTree : left;
    }
}