使用一个哑节点记录信息,以便使用到next的信息
def connect(self, root):
"""
:type root: Node
:rtype: Node
"""
if not root:
return
stack = [root]
while stack:
n = len(stack)
for i in range(n):
tmp = stack.pop(0)
if i == n-1:
tmp.next = None
else:
tmp.next = stack[0]
if tmp.left:
stack.append(tmp.left)
if tmp.right:
stack.append(tmp.right)
return root