Leetcode116. 在每个节点中填充下一个右指针 (Python3解法）

最新推荐文章于 2025-06-06 22:05:56 发布

原创最新推荐文章于 2025-06-06 22:05:56 发布 · 221 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

16 篇文章

订阅专栏

本文介绍了一种算法，用于解决完美二叉树中节点的Next指针连接问题。通过层次遍历的方式，利用队列实现节点间的连接，确保每个节点的Next指针指向其右侧的相邻节点。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

难度：Medium ?：1119

题目

给定一棵完美二叉树（所有叶子在同一层，每个父节点都有两个子节点），树的节点的定义中包含一个next指针，要做的事情是将每个节点的next指针指向下一个右边的节点。

示意图

思路

就是二叉树层次遍历问题，借助一个队列即可解决。

"""
# 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 not root:
            return root
        queue = [root]
        while queue:
            count = len(queue)  #这一层的节点数
            while count>0:
                node1 = queue.pop(0)
                count -= 1
                if node1.left:
                    queue.append(node1.left)
                if node1.right:
                    queue.append(node1.right)
                if count > 0: #每一层的最后一个无需修改next指针
                    node1.next = queue[0]
        return root