LeetCode 114 二叉树展开为链表 Java

114. 二叉树展开为链表

题目

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

例如，给定二叉树

    1
   / \
  2   5
 / \   \
3   4   6

将其展开为：

1
 \
  2
   \
    3
     \
      4
       \
        5
         \
          6

题解

​ 题目本身不复杂，也不知道为什么是中等难度，题意就是将二叉树原地展开一个单链表，从图看出应该沿着是节点的右指针增加节点，先展开左支，在展开右支，将根节点的右指针指向左支的头节点，最后将链表尾节点的右指针指向展开后的右支头节点。可能是因为原地展开导致了一眼看出来用栈或队列的答案出错。

class Solution {
    public void flatten(TreeNode root) {
        if(root==null) return;
        //先展开左支为链表
        flatten(root.left);
        //在展开右支
        flatten(root.right);
        TreeNode temp = root.right;
        root.right = root.left;
        //记得清空左支
        root.left = null;
        while(root.right!=null){
            root = root.right;
        }
        root.right = temp;

    }
}