JZ36 二叉搜索树与双向链表

题目

数据范围：输入二叉树的节点数 0 \le n \le 10000≤n≤1000，二叉树中每个节点的值 0\le val \le 10000≤val≤1000
要求：空间复杂度O(1)O(1)（即在原树上操作），时间复杂度 O(n)O(n)

注意:
1.要求不能创建任何新的结点，只能调整树中结点指针的指向。当转化完成以后，树中节点的左指针需要指向前驱，树中节点的右指针需要指向后继
2.返回链表中的第一个节点的指针
3.函数返回的TreeNode，有左右指针，其实可以看成一个双向链表的数据结构
4.你不用输出双向链表，程序会根据你的返回值自动打印输出

输入描述：
二叉树的根节点

返回值描述：
双向链表的其中一个头节点。
示例1

输入： {10,6,14,4,8,12,16}
返回值： From left to right are:4,6,8,10,12,14,16;From right to left are:16,14,12,10,8,6,4;
说明： 输入题面图中二叉树，输出的时候将双向链表的头节点返回即可。

代码

public class Solution {
    public TreeNode Convert(TreeNode pRootOfTree) {
        //把二叉搜索树进行中序遍历，按照中序遍历的结果来构造双向链表，
        //使用left指向前一个节点，使用right指向后一个节点
        if(pRootOfTree == null){
            return null;
        }
        if(pRootOfTree.left == null && pRootOfTree.right == null){
            return pRootOfTree;
        }
        //递归处理左子树，得到左子树的链表的头节点
        TreeNode leftHead = Convert(pRootOfTree.left);
        TreeNode leftTail = leftHead;
        //把当前节点尾插到左侧链表的末尾
        //需要先找到左侧链表的末尾节点，才能进行尾插
        while(leftTail!= null && leftTail.right != null ){
            leftTail = leftTail.right;
        }
        //尾插
        if(leftHead != null){
       leftTail.right = pRootOfTree;
        pRootOfTree.left = leftTail;
        }
        //递归处理右子树
        TreeNode rightHead = Convert(pRootOfTree.right);
        //把当前节点插入到右侧节点的头部
        if(rightHead != null){
            rightHead.left = pRootOfTree;
            pRootOfTree.right = rightHead;
        }
        return leftHead != null ? leftHead : pRootOfTree;
    }


}