LeetCode114 -- 二叉树展开为链表

题目描述

给定一个二叉树，原地将它展开为链表。

例如，给定二叉树

    1
    /  \
  2    5
 /   \    \

3   4     6
将其展开为：

1
 \
  2
   \
    3
     \
      4
       \
        5
         \
          6

解题思路

用递归的方式 分成四步

1 首先将根节点的左子树变成链表

2 其次将根节点的右子树变成链表

3 最后变成链表的左子树接到根节点上

4 将变成链表的右子树放在变成链表的左子树的最右边 

这就是一个递归的过程，递归的一个非常重要的点就是：不去管函数的内部细节是如何处理的，我们只看其函数作用以及输入与输出。对于函数flatten来说：

函数作用：将一个二叉树，原地将它展开为链表 ，

                 也就是对于每一次的递归flatten(当前root)完全结束的时候，都完成把当前root子树展开成链表

输入：树的根节点

输出：无

class Solution {
    public void flatten(TreeNode root) {
        // 这个题说的不清不楚的 猜测是按照前序遍历的顺序展开
        // 原地就是不能建立新的节点 就是用树的right指针作为链表的next指针
        if (root == null){
            return;
        }
        // 1 将左子树展开
        flatten(root.left);
        // 2 将右子树展开
        flatten(root.right);

        TreeNode tmp = root.right; //暂时保存
        // 3 把左链表接到root的右侧
        root.right = root.left;
        root.left = null; //别忘了左子树置空

        // 遍历到最右 接保存的右链表
        while (root.right != null){
            root = root.right;
        }
        root.right = tmp;
    }
}