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

Marshal～

于 2022-03-15 15:07:07 发布

阅读量1.3k

点赞数

分类专栏：刷题文章标签： leetcode 链表算法

本文链接：https://blog.csdn.net/M_arshal_/article/details/123502882

版权

Python collections deque 树结构节点连接

关键词由CSDN通过智能技术生成

刷题专栏收录该内容

16 篇文章 0 订阅

订阅专栏

在这里插入图片描述
这里用到的知识点主要是python中collections库中关于队列的使用。

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


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