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
实现结果
实现结果 | 递归
实现结果 | 非递归
随时随地看视频
