【LeetCode】116. Populating Next Right Pointers in Each Node 解题报告(Python)

22 篇文章 0 订阅
5 篇文章 0 订阅

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

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

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

示例:
在这里插入图片描述
解题思路1:

这一题可以采用队列层序遍历的方法(指向下一个会有一个判断,不能时当前层的最后一个,及1,3,7,他们都直接指向空)

  1. 1进队列,1出队列,1指向队列下一个空,1左(2)进队,1右(3)进队
  2. 2出队列,2指向队列下一个3,2左(4)进队,2右(5)进队
  3. 3出队列,3指向队列下一个空,3左(6)进队,3右(7)进队
  4. 4出队列,4指向队列下一个5,3左(6)进队,3右(7)进队
  5. 队列有4567,然后,逐个出队列并指向下一个,最后7指向空,就完事了

提交代码1:(队列层序,Runtime: 64 ms, faster than 81.18 % )

class Solution:
    def connect(self, root: 'Node') -> 'Node':
        if not root:
            return root
        Q = collections.deque([root])
        while Q:
            size = len(Q)
            for i in range(size):
                node = Q.popleft()
                if i < size - 1:
                    node.next = Q[0]
                if node.left:
                    Q.append(node.left)
                if node.right:
                    Q.append(node.right)
        return root

解题思路2:

这一题不用队列也能做,每次让根节点的左节点指向右节点,如果根节点有下一个指向节点,那么根节点右节点指向根节点下一个节点的左节点
提交代码2:(Runtime: 64 ms, faster than 81.18 % )

class Solution:
    def connect(self, root: 'Node') -> 'Node':
        if not root:
            return root

        leftmost = root
        while leftmost.left:
            head = leftmost
            while head:
                head.left.next = head.right
                if head.next:
                    head.right.next = head.next.left
                head = head.next
            leftmost = leftmost.left
        return root
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值