LeetCode 114. 二叉树展开为链表 | python

114. 二叉树展开为链表


题目来源:力扣(LeetCode)
https://leetcode-cn.com/problems/flatten-binary-tree-to-linked-list

题目


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

例如,给定二叉树

    1
   / \
  2   5
 / \   \
3   4   6

将其展开为:

1
 \
  2
   \
    3
     \
      4
       \
        5
         \
          6

解题思路


思路: 递归,非递归

递归

我们先观察例子,可以发现,左子树展开成链表连接在根节点,而右子树展开链表是紧跟在左子树展开的链表后面。这里使用递归的方法来解决。

使用后序遍历(递归)的具体做法:

  • 先找到左子树的最右节点;
  • 当找到左子树的最右节点时,令其指向根的右子树;
  • 此时,再让根的右子树,指向根节点的左子树;
  • 最后,令根节点的左子树为 None,循环直至右子树为空。

具体的代码见【代码实现 # 递归】

非递归

在前面我们知道,其实递归的思路,使用了额外的空间。这里,我们使用非递归的方法来尝试解决。(具体做法同上)

具体的代码见【代码实现 # 非递归】

代码实现


# 递归
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def flatten(self, root: TreeNode) -> None:
        """
        Do not return anything, modify root in-place instead.
        """
        def unflod(node):
            if not node:
                return None

            unflod(node.left)
            unflod(node.right)

            # 后序遍历
            if node.left:
                pre = node.left
                # 找到左子树的最右子节点,用以连接根的右子树
                while pre.right:
                    pre = pre.right
                # 当找到左子树的最右子节点,令其指向根的右子树
                pre.right = node.right
                # 然后令根的右子树指向根的左子树,令根的左子树为空
                node.right = node.left
                node.left = None
            # 循环
            node = node.right

        unflod(root)

# 非递归
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def flatten(self, root: TreeNode) -> None:
        """
        Do not return anything, modify root in-place instead.
        """
        while root:
            if root.left:
                pre_node = root.left
                # 同样先找到左子树的最右节点
                while pre_node.right:
                    pre_node = pre_node.right
                # 最右节点指向根节点的右子树
                pre_node.right = root.right
                # 根的右子树指向根的左子树,同时置空左子树
                root.right = root.left
                root.left = None
            root = root.right

实现结果


实现结果 | 递归

实现结果|递归

实现结果 | 非递归

实现结果 | 非递归

欢迎关注


公众号 【书所集录

阅读 42

推荐阅读