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 = []
输出：[]

题解

const flatten = (root) => {
  const helper = (root) => {       // 将当前子树转成一个单链表
    if (root == null) return null; // 遍历到null节点 返回null节点
    if (root.right) {              // 先生成右子树的单链表
      helper(root.right);
    }
    if (root.left) {               // 如果有左子树，生成单链表然后搬运过去
      const leftFirst = helper(root.left); // 生成单链表，并获取头结点
      //此时就默认左链表是一个向右倾斜的单链表了
      let leftLast = leftFirst;    // leftEnd是单链表的尾节点
      while (leftLast.right) {     // 一直找右节点，获取到单链表的尾节点
        leftLast = leftLast.right;
      }
      leftLast.right = root.right; // 尾节点后面接左子树展平后的单链表
      //根节点是最小的，所以根节点不动，把左子树变成的链表插到根和右链表之间
      root.right = leftFirst;      // 根节点的right改成leftFirst
      root.left = null;            // root.left置为null
    }
    return root; // 返回出当前子树转成的单链表
  };
  helper(root);  // 原地修改，不用返回
};

笔记：

1. 具体思路
   右子树转成的单链表，只要获取它的头结点，等左子树生成的链表生成好，接到它的尾节点
   左子树生成的单链表，处理稍微复杂一点

   • 要获取它的头结点，它要接到根节点的right
   • 要获取它的尾节点，通过一直找右节点，找到尾节点，供连接

   左子树生成的链表两端都接好后，root.left 要置为 null，不然 root 还拖着个左子树