填充每个节点的下一个右侧节点指针的思路探讨与源码
填充每个节点的下一个右侧节点指针的题目如下图,该题属于树和搜索类型的题目,主要考察对于递归方法的使用和树本身遍历方法的理解。本文的题目作者想到2种方法,分别是递归方法和层次遍历方法,其中递归方法使用Java进行编写,而层次遍历方法使用Python进行编写,当然这可能不是最优的解法,还希望各位大佬给出更快的算法。
本人认为该题目可以使用递归方法解决,首先判断树是否为空,如果是空则直接返回空。然后判断左节点是否不为空,如果是就开始继续遍历,将左节点的连接到右节点,再将右节点的连接设置为空,判断节点的连接是否不为空,如果是就将右节点的连接到下一个节点的左节点。然后继续用左节点和右节点进行递归调用,最终到递归结束并返回结果。那么按照这个思路我们的Java代码如下:
#喷火龙与水箭龟
class Solution {
public Node connect(Node root) {
if(root == null){
return root;
}
if(root.left != null){
root.left.next = root.right;
root.right.next = null;
if(root.next !=null){
root.right.next = root.next.left;
}
connect(root.right);
connect(root.left);
}
return root;
}
}
显然,我们看到递归的方法的效果不错,同时还可以使用层次遍历的方法。首先判断阶段是否为空,如果是就直接返回。然后初始化一个队列,将根节点添加到这个队列里面。开始遍历队列,记录队列的大小,然后遍历当前这一层的所有节点,并从队列的首部取出元素,进行连接的操作,如果左节点或者右节点不为空的话,那么将左节点或者右节点加入到队列里面。最终等到队列为空的时候,就结束循环并返回最终的结果。所以按照这个思路就可以解决,下面是Python代码部分:
#喷火龙与水箭龟
class Solution:
def connect(self, root: 'Node') -> 'Node':
if not root:
return root
queueList = collections.deque([root])
while queueList:
queueLen = len(queueList)
for kr in range(queueLen):
vertex = queueList.popleft()
if(kr < queueLen - 1):
vertex.next = queueList[0]
if(vertex.left):
queueList.append(vertex.left)
if(vertex.right):
queueList.append(vertex.right)
return root
从结果来说Java版本的递归方法的效率不错,同样的Python版本的分层遍历方法的速度也还可以,但应该是有更多的方法可以进一步提速的,希望朋友们能够多多指教,非常感谢。