LeetCode114. 二叉树展开为链表

5星上将

于 2020-08-02 14:58:56 发布

阅读量219

点赞数

分类专栏： LeetCode

本文链接：https://blog.youkuaiyun.com/FiveStarsGeneral/article/details/107744875

版权

LeetCode 专栏收录该内容

50 篇文章

订阅专栏

题目描述

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

例如，给定二叉树
在这里插入图片描述

将其展开为：

在这里插入图片描述

思路

思路很简单，按照例子给的，我们可以先前序访问二叉树节点并将其存到一个数组里，然后再将数组改成一个相当于是只含有右子树的二叉树

代码

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution(object):
    def flatten(self, root):
        """
        :type root: TreeNode
        :rtype: None Do not return anything, modify root in-place instead.
        """
        if not root:
            return
        linklist = []
        def preOrderList(root):
            if root:
                linklist.append(root)
                preOrderList(root.left)
                preOrderList(root.right)
        preOrderList(root)
        for i in range(1,len(linklist)):
            curNode,preNode = linklist[i],linklist[i-1]
            preNode.left = None
            preNode.right = curNode