Leetcode刷题--114. 二叉树展开为链表

题解参考: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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

示例
Leetcode刷题--114. 二叉树展开为链表_第1张图片
根据题解中多次提到的,先给函数下一个定义:
给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了,递归上去它自然会实现,这种思想,需要时间培养一下。

你可能感兴趣的:(leetcode)