文巾解题 116. 填充每个节点的下一个右侧节点指针

Posted 2021-09-05 UQI-LIUWJ

1 题目描述

 ​​​​​​​

2 解题思路：广搜

 记录每一层的节点数量，每次遍历一层，当遇到这一层最后一个节点时，它的next指向None，其余节点则指向后面一个

class Solution:
    def connect(self, root: 'Node') -> 'Node':
        if(root==None):
            return None
        else:
            lst=[root]
            while(lst):
                n=len(lst) 
                for i in range(n):
                    tmp=lst.pop(0)

                    if(i<n-1):
                        tmp.next=lst[0]
                    else:
                        tmp.next=None
                    if(tmp.left!=None):
                        lst.append(tmp.left)
                    if(tmp.right!=None):
                        lst.append(tmp.right)
                        
        return root