算法--二叉树展开

Leetcode 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]

展开的顺序是二叉树的先序遍历的顺序。  把左子树不断的插入到右子树，原来的右子树放嫁接过来的左子树的最右边。

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21

public void flatten(TreeNode root) {     while (root != null) {         //左子树为 null，直接考虑下一个节点         if (root.left == null) {             root = root.right;         } else {             // 找左子树最右边的节点             TreeNode pre = root.left;             while (pre.right != null) {                 pre = pre.right;             }             //将原来的右子树接到左子树的最右边节点             pre.right = root.right;             // 将左子树插入到右子树的地方             root.right = root.left;             root.left = null;             // 考虑下一个节点             root = root.right;         }     } }

考虑把左子树嫁接到右子树，让原来的右子树接到左子树最后。不断的递归最终完成先序遍历的同时完成了单链表的构建。

?

1 2	if (root.left == null) {            root = root.right;<br>考虑root当前的左子树是否为null，如果为null，就只能遍历右边。根节点出发，左子树没有，右子树也可能出现需要把右子树的左孩子拼到右侧的情况<br><br>

?

1 2 3 4 5

// 找左子树最右边的节点            TreeNode pre = root.left;            while (pre.right != null) {                pre = pre.right;            } <br>在这里，保存左子树的节点pre.   如果左子树有右孩子，就保存左子树的右孩子为pre。   这里保留pre是为了让root节点的右子树拼到root节点的左子树的后面。

?

1 2 3

<br> //将原来的右子树接到左子树的最右边节点             pre.right = root.right;             // 将左子树插入到右子树的地方

右子树拼到左子树的pre节点（要连接root右子树的节点）的右孩子节点。
    root.right = root.left;

?

1 2 3

<em id="__mceDel">左子树插入到右子树的地方，这时候，右子树的节点断开。<br><br>root.left = null;<br>清空右子树的节点。<br> // 考虑下一个节点。root节点的下一个右节点作为root节点依次递归下去。 root = root.right;</em>