LeetCode 热题 HOT 100 Java题解——114. 二叉树展开为链表

114. 二叉树展开为链表

题目：

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

示例：

例如，给定二叉树

    1
   / \
  2   5
 / \   \
3   4   6
将其展开为：

1
 \
  2
   \
    3
     \
      4
       \
        5
         \
          6

前驱节点

笔者认为本题的关键在于原地，很容易想到的先序遍历一遍得出顺序，再按照顺序修改二叉树不能属于原地。因此，笔者认为最好的做法还是使用寻找前驱节点的办法：

1. 从根开始遍历节点，如果该节点的左子树为空，那么可以直接遍历其右节点。
2. 如果左子树不为空，那么就需要把当前的右子树放到左子树的最后一个节点后面，也就是左子树的最右节点，然后将当前节点的$right$改为$left$，$left$置为空。同样继续遍历右节点。

class Solution {
    public void flatten(TreeNode root) {
        while(root != null) {
            if (root.left != null) {
                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;
        }
    }
}

复杂度分析

• 时间复杂度：$O(n)$

  展开为单链表的过程中，需要对每个节点访问一次，在寻找前驱节点的过程中，每个节点最多被额外访问一次。

• 空间复杂度：$O(1)$

  使用常量级额外空间。