欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

【python/M/114】Flatten Binary Tree to Linked List

程序员文章站 2022-04-27 08:49:57
...

题目

【python/M/114】Flatten Binary Tree to Linked List

基本思路

首先呢,我们要来理解题意,按先序遍历的顺序将一棵二叉树变成一个类似的单链表。
我们这么做:首先对树进行先序遍历同时将所有节点错放在一个数组中,数组中元素类型为一个个TreeNode,然后遍历这个数组,将元素依次连接起来,记住要将左子树节点变成空。

实现代码

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def flatten(self, root):
        """
        :type root: TreeNode
        :rtype: void Do not return anything, modify root in-place instead.
        """
        # 充分理解展开后列表的这个形式,不是个单链表!
        ans = []
        def preOrder(root):
            if not root:
                return 
            ans.append(root)
            preOrder(root.left)
            preOrder(root.right)
        preOrder(root)
        p = root
        for i in range(1,len(ans)):
            p.left = None
            p.right = ans[i]
            p = ans[i]


相关标签: python leetcode