leetcode刷题250天(13)——116. 填充每个节点的下一个右侧节点指针
"""
# 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:
def connect(self, root):
if root:
Tovisit = [root]
else:
return root
while Tovisit:
len_ = len(Tovisit)
for i in range(len_):
cur = Tovisit.pop(0)
if cur and i < len_ - 1:
cur.next = Tovisit[0]
if cur.left:
Tovisit.append(cur.left)
Tovisit.append(cur.right)
return root