LeetCode(114) Flatten Binary Tree to Linked List解题报告

最新推荐文章于 2019-12-25 11:35:29 发布

Ssunsets

最新推荐文章于 2019-12-25 11:35:29 发布

阅读量525

点赞数

分类专栏： LeetCode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/Ssunsets/article/details/50290943

版权

LeetCode 专栏收录该内容

32 篇文章

订阅专栏

本文详细介绍了如何通过迭代而非递归的方式，将二叉树扁平化为链表，有效避免了空间复杂度的增加，提高了时间效率。包括两种解法的对比与分析。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given a binary tree, flatten it to a linked list in-place.

For example,
Given

The flattened tree should look like:

解题思路：
这题花了不少时间，一开始中序遍历二叉树，用数组存节点，然后重新生成树，结果超时，考虑到重新生成的空间复杂度和直接生成结点空间复杂度是一样的，但是时间复杂度会增加n，所以直接生成二叉树，然后将更新根节点，这里有一点要注意的是，注意避免覆盖生成的节点，如果用这种思路发现输入和输出一直一样的话，那说明没弄明白指针和引用的区别，或者说没弄清楚java都是值传递。

public class Solution {
    TreeNode pro;
    public void flatten(TreeNode root) {
        TreeNode head;
        if(root == null)
            return;
        pro = new TreeNode(0);
        head = pro;
        inorder(root);
        root.left = null;
        root.right = head.right.right;
    }
    public TreeNode inorder(TreeNode root){
        TreeNode temp = new TreeNode(root.val);
        pro.right = temp;
        pro = pro.right;
        if(root.left != null)
            pro.right = inorder(root.left);
        while(pro.right != null)
            pro = pro.right;
        if(root.right != null)
            pro.right = inorder(root.right);
        return temp;
    }
}

下面附上一种更优质的解法

转自博客:http://blog.youkuaiyun.com/sbitswc/article/details/26540269

分析发现，左子树节点都在右子树前面，左子树的最后一个节点，在右子树的第一个节点之前，通过迭代的把左子树插入到右子树，并在第一个节点之前，生成FBT。

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;
        }
    }