leetcode 117. 填充每个节点的下一个右侧节点指针 II

117. 填充每个节点的下一个右侧节点指针 II
给定一个二叉树

struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL。

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

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

示例:

输入:root = [1,2,3,4,5,null,7]
输出:[1,#,2,3,#,4,5,7,#]
解释:给定二叉树如图 A 所示,你的函数应该填充它的每个 next 指针,以指向其下一个右侧节点,如图 B 所示。

在这里插入图片描述


树和图有两种基本遍历方法。一种是深度优先遍历,例如:每次只遍历一个分支;另外一种是广度优先遍历,例如:先遍历完一层再进入下一层。树的深度优先遍历又可以分为先序遍历 preorder、中序遍历 inorder 和后序遍历 postorder。树的广度优先遍历基于节点的层级 level 概念。一个节点的层级取决于该节点的深度或者到根节点的距离。需要先遍历完同一层级的所有节点,才能进入下一层级。

很明显,此问题应该使用广度优先遍历。使用广度优先遍历,可以将同一层级的所有节点连接起来。

###
#和上面一道题类似,只不过这里需要考虑跳连的情况,不是完美二叉树,需要对每个节点的子节点个数进行判断
#这里使用的是迭代的方法
###
class Solution:
    def connect(self, root: 'Node') -> 'Node':
        def is_valid(root):
            if not root: return #节点不存在
            if root.left: return root.left#节点左边存在,如果左右都存在还是优先返回左
            elif root.right:return root.right#之后节点右边存在,返回右
            else: return root#节点存在,但是没有子节点,返回本身

        if not root:#如果节点为空,直接返回
            return 
        pre = root#设定初始的根节点
        while pre:#当根节点存在的时候继续,这个点是每一行的最左边的点
            cur = pre#对每一行的节点进行连接
            while cur:#当每一行的计算节点不为空时
                temp = is_valid(cur)#判断当前节点的状态,空,无子节点,有左子节点,有有子节点,有两个节点,这五种情况
                if temp == cur.left or temp == cur.right: #存在一个或者两个子节点时,可以进行连接
                    if cur.left and cur.right:#如果该节点下存在两个节点,那么首先连接左右节点,右节点再与后面的连接
                        cur.left.next = cur.right
                        pre_cur = cur#在子右节点连接后面的节点时,我们需要找到后面最开始出现的子节点
                        while cur.next and cur.next == is_valid(cur.next):#这一步就是跳过一些没有子节点的当前层节点
                            cur = cur.next
                        pre_cur.right.next = is_valid(cur.next)#直到找到最后的空null,或者是出现的子节点,进行连接
                    else:#只有一个子节点的时候,也是先找后面一个子节点出的地方
                        while cur.next and cur.next == is_valid(cur.next):
                            cur = cur.next
                        temp.next = is_valid(cur.next)#将前面的某个子节点与后面的节点连接
                cur = cur.next#节点连接过后
            #和上道题目一样,这里应该是确定每一层的起始位置,首先看本次节点是否有子节点,没有就找next,后面的一个节点有不有子节点
            pre = is_valid(pre) if is_valid(pre) == pre.left or is_valid(pre) == pre.right else is_valid(pre.next)
        return root

这里放一个大佬的代码:
首先是对每一层进行遍历,

"""
# 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':
        cur = root
        head = None
        tail = None
        while cur:
            while cur:
                if cur.left:
                    if not head:#如果还没有找到头,初始状态,只有一个数,所以头尾都等于这个数
                        head = cur.left
                        tail = cur.left
                    else:#如果已经有头了,那么加入列表,尾部后移一位
                        tail.next = cur.left
                        tail = tail.next
                if cur.right:
                    if not head:#同上,如果还没有头,加入列表,头尾都是一样的
                        head = cur.right
                        tail = cur.right
                    else:#如果已经由头,将右节点加入,尾标后移一位
                        tail.next = cur.right
                        tail = tail.next
                cur = cur.next#每一层的节点右移
            #这一步很巧妙,因为下一层的起始点和当前层的起始点有关,这里保存的是上一层的起始点位置
            cur = head
            head = None#再把头尾置None
            tail = None
        return root

作者:powcai
链接:https://leetcode-cn.com/problems/populating-next-right-pointers-in-each-node-ii/solution/die-dai-by-powcai/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值