自学内容网 自学内容网

leetcode-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]

思路

1)先序遍历二叉树到一个list里

2)遍历list,前一个节点为根,后一个节点为右孩子,在置左孩子为None

# 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 __init__(self):
        self.res = []
    def pretraverse(self, root):
        if not root:
            return None
        self.res.append(root)
        self.pretraverse(root.left)
        self.pretraverse(root.right)
    def flatten(self, root):
        """
        :type root: TreeNode
        :rtype: None Do not return anything, modify root in-place instead.
        """
        self.pretraverse(root)
        for i in range(1, len(self.res)):
            pre,cur = self.res[i-1], self.res[i]
            pre.left = None
            pre.right = cur
        

原文地址:https://blog.csdn.net/m0_38098373/article/details/140683598

免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!