题解参考:https://labuladong.gitbook.io/algo/shu-ju-jie-gou-xi-lie/shou-ba-shou-shua-er-cha-shu-xun-lian-di-gui-si-wei/er-cha-shu-xi-lie-1
写树的递归,首先要想清楚使用二叉树的哪种遍历方式
本题需要对最下面的子节点进行操作,而且我们操作的只能是父节点,注意,操作父节点,所以,符合要求的只能是后序遍历。
树的递归操作,核心都是,在哪步操作根节点(就是合理选择前中后序遍历)
给你二叉树的根结点 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]
提示:
树中结点数在范围 [0, 2000] 内
-100 <= Node.val <= 100
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/flatten-binary-tree-to-linked-list
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
示例
根据题解中多次提到的,先给函数下一个定义:
给flatten函数输入一个节点root,那么以root为根的二叉树就会被拉平为一条两边。
所以操作大致分为:
1、将root的左子树和右子树拉平
2、将root的右子树接到左子树下方,然后将整个左子树作为右子树
class Solution(object):
def flatten(self, root: TreeNode) -> None:
if root is None:
return
# 要清楚需要使用二叉树的哪种遍历方式
# 本题需要先更改子节点,所以使用后序遍历框架
# 而且最重要的是,我们都是在递归弹回来的时候进行操作
# 此时操作的不是子节点,而是根节点。重要,而是根节点。
# 子节点没法访问父节点,所以,我们操作的是根节点
# 在图中,操作的是,1 2 5 节点,这就是后序遍历。
# 其他顺序的遍历核心也是,在哪步操作根节点,根节点,根节点。
self.flatten(root.left)
self.flatten(root.right)
# 后序遍历位置
left = root.left
right = root.right
# 将左子树变成右子树
root.left = None
root.right = left
'''
疑问代码
root.right.right = right
roo.right.right 不就是更改到右边之后的left吗
为什么运行会报错,说Nonetype类型没有attribute right呢
'''
# 将原先的右子树接到当前右子树的末端
temp = root
while temp.right:
temp = temp.right
temp.right = right
其中有一个疑问
以及思想很重要,递归思想,我们只需要写基本步骤就行,不必关心它是怎么实现的。
本题就是,将左子节点变成右子节点,然后右子节点接到左子节点的右边。就ok了,递归上去它自然会实现,这种思想,需要时间培养一下。