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

最新推荐文章于 2025-09-03 09:21:50 发布

緈福的街口

最新推荐文章于 2025-09-03 09:21:50 发布

阅读量327

点赞数 5

CC 4.0 BY-SA版权

文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/qq_38473254/article/details/150871478

LeetCode 专栏收录该内容

124 篇文章

订阅专栏

文章目录

- 题目
- 题解

题目

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

给定一个二叉树：

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

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

在这里插入图片描述

题解

层序遍历
每一层指定一个prev


 if prev:
     prev.next = node

 prev = node

"""
# Definition for a Node.
class Node(object):
    def __init__(self, val=0, left=None, right=None, next=None):
        self.val = val
        self.left = left
        self.right = right
        self.next = next
"""

class Solution(object):
    def connect(self, root):
        """
        :type root: Node
        :rtype: Node
        """
        if not root:
            return root
        res = []
        queue = collections.deque()
        queue.append(root)
        while queue:
            prev = None
            for i in range(len(queue)):
                node = queue.popleft()
                if prev:
                    prev.next = node

                prev = node
                
                if node.left:
                    queue.append(node.left)
                if node.right:
                    queue.append(node.right)
	        if prev:
	                prev.next = None

        return root