Flatten Binary Tree to Linked List -- LeetCode

二叉树展平为链表

最新推荐文章于 2022-07-27 15:08:45 发布

最新推荐文章于 2022-07-27 15:08:45 发布 · 193 阅读

文章标签：

#数据结构与算法

本文介绍了一种将二叉树按照先序遍历顺序展平为链表的方法，该链表由树的节点构成，每个节点的左指针为空，右指针指向下一个节点。使用递归实现，并在过程中注意保存右子节点以防丢失。

原题链接: http://oj.leetcode.com/problems/flatten-binary-tree-to-linked-list/
这是树的题目，要求把一颗二叉树按照先序遍历顺序展成一个链表，不过这个链表还是用树的结果，就是一直往右走（没有左孩子）来模拟链表。老套路还是用递归来解决，维护先序遍历的前一个结点pre，然后每次把pre的左结点置空，右结点设为当前结点。这里需要注意的一个问题就是我们要先把右子结点保存一下，以便等会可以进行递归，否则有可能当前结点的右结点会被覆盖，后面就取不到了。算法的复杂度时间上还是一次遍历，O(n)。空间上是栈的大小，O(logn)。代码如下：

public void flatten(TreeNode root) {
    ArrayList<TreeNode> pre = new ArrayList<TreeNode>();
    pre.add(null);
    helper(root, pre);
}
private void helper(TreeNode root, ArrayList<TreeNode> pre)
{
    if(root == null)
        return;
    TreeNode right = root.right;
    if(pre.get(0)!=null)
    {
        pre.get(0).left = null;
        pre.get(0).right = root;
    }
    pre.set(0,root);
    helper(root.left, pre);
    helper(right, pre);
}

树的题目讨论得比较多了，主要思路就是递归，考虑好递归条件和结束条件，有时候如果递归过程会对树结构进行修改的话，要先保存一下结点。