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]

进阶：你可以使用原地算法（O(1) 额外空间）展开这棵树吗？

·解题思路

最简单的思路就是遍历每一个结点，将右节点挂到左节点最右端的下端，然后再将左节点挂到最上端结点的右端，左节点变为null；这是从上到下的操作。

Java代码

class Solution {
    public void flatten(TreeNode root) {
        while(root != null){
            if(root.left == null) root = root.right ;
            else{
                TreeNode pre = root.right;
                while(pre.right != null) pre = pre.right;
                pre.right = root.right;
                root.right = root.left;
                root.left = null;
                root = root.right;
            }
        }
    }}

·解题思路2

从下往上走，对于每一个结点，将其左右结点压入栈中保存，再弹出依次挂到右端上；这需要保存一个pre指针来辅助。

Java代码

    public TreeNode pre = null;
    public void flatten(TreeNode root) {
        if(root == null) return ;
        Stack<TreeNode> stack = new Stack<>();
        stack.push(root);
        while(!stack.isEmpty()) {
            TreeNode temp = stack.pop();
            if(pre != null){
                pre.right = temp;
                pre.left = null;
            }
            if(temp.left != null) stack.push(temp.left);
            if(temp.right != null) stack.push(temp.right);
            pre = temp;
        }
    }