leetcode-114 二叉树展开为链表

题目描述：给你二叉树的根结点 root ，请你将它展开为一个单链表：

• 展开后的单链表应该同样使用 TreeNode ，其中 right 子指针指向链表中下一个结点，而左子指针始终为 null 。
• 展开后的单链表应该与二叉树 先序遍历 顺序相同。

代码实现

递归实现

class Solution {
    public void flatten(TreeNode root) {
        convertTree2List(root);
    }

    public TreeNode convertTree2List(TreeNode root) {
        //1. 判空
        if (root == null || (root.left == null && root.right == null)){
            return root;
        }

        //2. 转换左子树
        TreeNode leftHeadNode = convertTree2List(root.left);

        //3. 转换右子树
        TreeNode rightHeadNode = convertTree2List(root.right);

        //4. 找到左子树的叶子节点
        TreeNode leftTailNode = leftHeadNode;
        while (leftTailNode != null && leftTailNode.right != null){
            leftTailNode = leftTailNode.right;
        }

        //5. 拼接左右子树与根节点
        if (leftTailNode == null){
            root.right = rightHeadNode;
            root.left = null;
        }else {
            root.right = leftHeadNode;
            leftTailNode.right = rightHeadNode;
            root.left = null;
        }
        return root;
    }
}