leetcode:填充每一个节点的下一个右侧节点(python)

乖乖的函数 2020-11-13 02:53:49
LeetCode 节点 一个 填充 每一个


1. 题目描述

给定一个完美二叉树,其所有叶子节点都在同一层,每个父节点都有两个子节点。二叉树定义如下:

struct Node {
int val;
Node *left;
Node *right;
Node *next;
}

填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL。

初始状态下,所有 next 指针都被设置为 NULL。

示例:
在这里插入图片描述

解释:给定二叉树如图 A 所示,你的函数应该填充它的每个 next 指针,以指向其下一个右侧节点,如图 B 所示。

提示:
你只能使用常量级额外空间。
使用递归解题也符合要求,本题中递归程序占用的栈空间不算做额外的空间复杂度。

2. 思路

由于是完美二叉树,因此不需要担心左子树存在右子树不存在的情况,也不用担心某一节点有子树,而同层的节点没有子树的情况发生。即要有都有,要没有都没有。

2.1 递归

对于递归的话,一定要建立相邻节点的右子树和左子树的联系,要不然,每个节点的右子树的next都会是None,即如果父节点有Next节点的话,父节点的右子树的next是父节点next节点的左子树,这样才能建立联系。

2.1 python代码

"""
# Definition for a Node.
class Node:
def __init__(self, val, left, right, next):
self.val = val
self.left = left
self.right = right
self.next = next
"""
class Solution:
def connect(self, root: 'Node') -> 'Node':
if root == None:
return None
if root.left == None:
return root
root.left.next = root.right
if root.next: # 如果root.next存在的话
root.right.next = root.next.left
self.connect(root.left)
self.connect(root.right)
return root

2.2 非递归

非递归的话,其实可以通过队列,来进行层次遍历,但是这样的话,会超出空间复杂度的限制。
每一层节点的next节点的设置都是从最左边开始的,因此需要设置一个指示每一层最左边父节点的指针preNode。使得每次循环都是从preNode.left开始的。对于某一层来说,从左到右依次设置,每一次都需要检查父节点有没有next,要是有的话,建立父节点的右子节点与父节点next节点左子树之间的联系。

2.2 python 代码

"""
# Definition for a Node.
class Node:
def __init__(self, val, left, right, next):
self.val = val
self.left = left
self.right = right
self.next = next
"""
class Solution:
def connect(self, root: 'Node') -> 'Node':
if root == None:
return None
preNode = root # 记录最左节点的信息
curNode = root # 记录当前子节点的父节点的信息随时更新
while True:
if curNode.left:
curNode.left.next = curNode.right
if curNode.next: # 检查父节点有没有next,有的话建立联系,
# 也说明这一层还没修改完,修改curNode为其next节点
curNode.right.next = curNode.next.left
curNode = curNode.next
else: #如果没有next节点了,说明这一层已经修改完了,
# 修改curNode到下一层的最左子节点,同时修改preNode,
curNode = preNode.left
preNode = curNode
else: # 如果curNode.left不存在,说明已经是最后一层了,打断循环
break
return root
版权声明
本文为[乖乖的函数]所创,转载请带上原文链接,感谢
https://blog.csdn.net/ggdhs/article/details/92979459

  1. 利用Python爬虫获取招聘网站职位信息
  2. Using Python crawler to obtain job information of recruitment website
  3. Several highly rated Python libraries arrow, jsonpath, psutil and tenacity are recommended
  4. Python装饰器
  5. Python实现LDAP认证
  6. Python decorator
  7. Implementing LDAP authentication with Python
  8. Vscode configures Python development environment!
  9. In Python, how dare you say you can't log module? ️
  10. 我收藏的有关Python的电子书和资料
  11. python 中 lambda的一些tips
  12. python中字典的一些tips
  13. python 用生成器生成斐波那契数列
  14. python脚本转pyc踩了个坑。。。
  15. My collection of e-books and materials about Python
  16. Some tips of lambda in Python
  17. Some tips of dictionary in Python
  18. Using Python generator to generate Fibonacci sequence
  19. The conversion of Python script to PyC stepped on a pit...
  20. Python游戏开发,pygame模块,Python实现扫雷小游戏
  21. Python game development, pyGame module, python implementation of minesweeping games
  22. Python实用工具,email模块,Python实现邮件远程控制自己电脑
  23. Python utility, email module, python realizes mail remote control of its own computer
  24. 毫无头绪的自学Python,你可能连门槛都摸不到!【最佳学习路线】
  25. Python读取二进制文件代码方法解析
  26. Python字典的实现原理
  27. Without a clue, you may not even touch the threshold【 Best learning route]
  28. Parsing method of Python reading binary file code
  29. Implementation principle of Python dictionary
  30. You must know the function of pandas to parse JSON data - JSON_ normalize()
  31. Python实用案例,私人定制,Python自动化生成爱豆专属2021日历
  32. Python practical case, private customization, python automatic generation of Adu exclusive 2021 calendar
  33. 《Python实例》震惊了,用Python这么简单实现了聊天系统的脏话,广告检测
  34. "Python instance" was shocked and realized the dirty words and advertisement detection of the chat system in Python
  35. Convolutional neural network processing sequence for Python deep learning
  36. Python data structure and algorithm (1) -- enum type enum
  37. 超全大厂算法岗百问百答(推荐系统/机器学习/深度学习/C++/Spark/python)
  38. 【Python进阶】你真的明白NumPy中的ndarray吗?
  39. All questions and answers for algorithm posts of super large factories (recommended system / machine learning / deep learning / C + + / spark / Python)
  40. [advanced Python] do you really understand ndarray in numpy?
  41. 【Python进阶】Python进阶专栏栏主自述:不忘初心,砥砺前行
  42. [advanced Python] Python advanced column main readme: never forget the original intention and forge ahead
  43. python垃圾回收和缓存管理
  44. java调用Python程序
  45. java调用Python程序
  46. Python常用函数有哪些?Python基础入门课程
  47. Python garbage collection and cache management
  48. Java calling Python program
  49. Java calling Python program
  50. What functions are commonly used in Python? Introduction to Python Basics
  51. Python basic knowledge
  52. Anaconda5.2 安装 Python 库(MySQLdb)的方法
  53. Python实现对脑电数据情绪分析
  54. Anaconda 5.2 method of installing Python Library (mysqldb)
  55. Python implements emotion analysis of EEG data
  56. Master some advanced usage of Python in 30 seconds, which makes others envy it
  57. python爬取百度图片并对图片做一系列处理
  58. Python crawls Baidu pictures and does a series of processing on them
  59. python链接mysql数据库
  60. Python link MySQL database