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

实现结果

---

实现结果 | 递归

实现结果 | 非递归

欢迎关注

---

公众号 【书所集录】