leetcode114. 二叉树展开为链表

chandelierds

已于 2023-02-05 22:29:26 修改

阅读量90

点赞数

分类专栏： Python 刷题 leetcode 文章标签：链表数据结构 leetcode

于 2019-07-02 23:18:29 首次发布

本文链接：https://blog.csdn.net/chandelierds/article/details/94493699

版权

leetcode 同时被 3 个专栏收录

123 篇文章 0 订阅

订阅专栏

刷题

98 篇文章 1 订阅

订阅专栏

Python

83 篇文章 1 订阅

订阅专栏

在这里插入图片描述

思路一

把该树进行前序遍历，然后再连接起来。

class Solution:
    def flatten(self, root: TreeNode) -> None:
        """
        Do not return anything, modify root in-place instead.
        """
        res = []
        if not root:
            return root
        def helper(root):
            if not root:
                return
            res.append(root.val)
            helper(root.left)
            helper(root.right)
        helper(root)
        size = len(res)
        for i in range(1, size):
            pre, cur = res[i - 1], res[i]
            pre.left = None
            pre.right = cur

思路二
递归

def flatten(self, root: TreeNode, pre = None) -> None:
        """
        Do not return anything, modify root in-place instead.
        """
        # 类似后序遍历
        def helper(root, pre):
            if not root: return pre
            # 记录遍历时候,该节点的前一个节点
            pre = helper(root.right, pre)
            pre = helper(root.left, pre)
            # 拼接
            root.right = pre
            root.left = None
            pre = root
            return pre
        helper(root, None)