114. 二叉树展开为链表

给你二叉树的根结点 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

题目明确说要与二叉树的前序遍历的顺序相同,那么很简单想到的暴力法就是将前序遍历存下来,然后一个一个放回树上。

class Solution:
    def flatten(self, root: TreeNode) -> None:
        """
        Do not return anything, modify root in-place instead.
        """
        # 暴力法
        result = []
        if not root:
            return []
        if not root.left and not root.right:
            return root
        def pre_order(node):
            result.append(node)
            if node.left:
                pre_order(node.left)
            if node.right:
                pre_order(node.right)
        
        pre_order(root)
        temp = root
        while result:
            temp.right = result.pop(0)
            temp.left = None
            temp = temp.right

这个方法简单,但是消耗比较大。
时间复杂度为O(N),空间复杂度为O(N)。

第二种方法是在前序遍历的基础上直接把树生成,也就是说,当前遍历到的节点为上一轮次遍历节点的右子节点,而左子节点置空。就会遇到问题:直接放过来的话会破坏树的结构,让树的遍历出现问题,故不能这么做。那么我们只能反过来,从末尾开始找,然后以末尾为右子节点再向上建树。
发现顺序是与前序遍历的倒序(并不是后序遍历)。
这里还有一个问题是,我们不能通过global来简单声明全局变量。global的用法,但是我们可以直接在类声明中声明变量。

# 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 __init__(self):
        self.pre = None

    def flatten(self, root: TreeNode) -> None:
        """
        Do not return anything, modify root in-place instead.
        """
        # pre = None
        # 前序遍历的逆序
        def order(node):
            # global pre
            if not node:
                return
            order(node.right)
            order(node.left)
            # 将节点接上去
            node.right = self.pre
            node.left = None
            self.pre = node
        
        order(root)

比较巧妙,用pre来记录当前节点的右子节点
时间复杂度是O(N)
空间复杂度是O(1)

第三种方法就是题目中要求的inplace法,要求只在原本的树上进行顺序修改,得到新的返回结果。

    1
   / \
  2   5
 / \   \
3   4   6

//1 的左子树插入到右子树的地方
    1
     \
      2         5
     / \         \
    3   4         6        
//将原来的右子树接到左子树的最右边节点
    1
     \
      2          
     / \          
    3   4  
         \
          5
           \
            6
            
 //2 的左子树插入到右子树的地方
    1
     \
      2          
       \          
        3       4  
                 \
                  5
                   \
                    6   
        
 //将原来的右子树接到左子树的最右边节点
    1
     \
      2          
       \          
        3      
         \
          4  
           \
            5
             \
              6         
  
  ......

作者:windliang
链接:https://leetcode-cn.com/problems/flatten-binary-tree-to-linked-list/solution/xiang-xi-tong-su-de-si-lu-fen-xi-duo-jie-fa-by--26/

上面的是大佬的示意图,很适合理解。那么接下来代码也很顺利写出来。

# 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 not root.left:
                root = root.right
            else:
                # 记录下right的情况,之后放在末尾
                temp = root.right
                # 把right换成left,left置空
                root.right = root.left
                root.left = None
                # root往下走
                root = root.right
                # 把temp放到末尾
                temp_ = root
                while temp_.right:
                    temp_ = temp_.right
                temp_.right = temp

时间复杂度为O(N)
空间复杂度为O(1)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值