栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

LeetCode 114. 二叉树展开为链表 | python

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

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  
实现结果
实现结果 | 递归

实现结果 | 非递归

转载请注明:文章转载自 www.mshxw.com
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号