Leetcode面试经典150题-114.二叉树展开为链表

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

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

示例 1：

输入：root = [1,2,5,3,4,null,6]
输出：[1,null,2,null,3,null,4,null,5,null,6]

示例 2：

输入：root = []
输出：[]

示例 3：

输入：root = [0]
输出：[0]

提示：

• 树中结点数在范围 [0, 2000] 内
• -100 <= Node.val <= 100

进阶：你可以使用原地算法（O(1) 额外空间）展开这棵树吗？

还算一道相对简单的问题，咱们直接实现进阶的解吧，不墨迹了，上代码

看不懂的请留言或者私信，收到第一时间解答

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    /**这个题大概想着应该和二叉树的先序遍历差不多，就是让自己的right指针指向自己左孩子的转换之后的头
    然后拿到左孩子的尾，左孩子的尾连接右孩子的头，大概估计是这么个思路，试试十分钟要是写不出来我再换方法*/
    public void flatten(TreeNode root) {
        process(root);
    }

    /**返回以root为头的树转换之后的的最后一个节点 */
    public TreeNode process(TreeNode root) {
        /**null当然返回null，要不然呢 */
        if(root == null) {
            return root;
        }
        /**如果没有左右子树，那它自己就是最后一个 */
        if(root.left == null && root.right == null) {
            return root;
        }
        /**先把右孩子拿出来，因为一会右指针要变了*/
        TreeNode right = root.right;
        /**如果左孩子为空，只转换右孩子 */
        if(root.left == null) {
            TreeNode rightLast = process(root.right);
            return rightLast;
        }
        /**到这里说明左节点不为空，那root的right指针应该指向root.left处理之后的结果*/
        TreeNode leftLast = process(root.left);
        /**它的right指针要连到左树的头，也就是左孩子 */
        root.right = root.left;
        root.left = null;
        /**如果右孩子为空，处理就结束了 */
        if(right == null) {
            return leftLast;
        }
        /**不为空找到右孩子的最后一个节点 */
        TreeNode rightLast = process(right);
        /**左边最右一个节点连接右边的头 */
        leftLast.right = right;
        /**返回最后一个节点，也就是右树的最后一个节点 */
        return rightLast;
    }
}