题意是说给你一棵二叉树,每个树节点中增加了一个next指针,指向了其同一层的相邻的右边的节点。
一开始由于考虑到使用层次遍历的方法,因此维护一个队列,队列中的每个节点记录其所属的深度,于是对同一层的节点使其next指针指向该节点在队列中的下一个节点就可以了。花费的时间复杂度是O(n),空间复杂度为O(n)。
然而由于题目中要求使用const space,于是考虑优化空间的方法。
无论如何是要进行层次遍历的,当我们进行第i层遍历的时候,其实第i-1层的next指针我们已经构建完成了,也就是说我们沿着上一层的next指针走就能遍历完上一层的所有节点。同时由于第i层的节点我们可以通过第i-1层的left、right指针进行访问,因此我们可以省略掉之前使用的队列,从而达到空间复杂度为O(1),时间复杂度为O(n)的解法。
程序中cur指向当前层的当前节点,head指向下一层的头指针,prev指向下一层的还未进行next指针设置的最后一个节点,也就是将要设置next指针的那个节点。不清楚的话可以看下代码:
# Definition for a binary tree node
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
# self.next = None
class Solution:
# @param root, a tree node
# @return nothing
def connect(self, root):
if root==None:
return
head = None
prev = None
cur = root
while cur!=None:
while cur!=None:
if cur.left!=None:
if prev==None:
head = cur.left
else:
prev.next = cur.left
prev = cur.left
if cur.right!=None:
if prev==None:
head = cur.right
else:
prev.next = cur.right
prev = cur.right
cur = cur.next
cur = head
prev = None
head = None