LeetCode刷题——填充每个节点的下一个右侧节点指针#116#Medium

填充每个节点的下一个右侧节点指针的思路探讨与源码
    填充每个节点的下一个右侧节点指针的题目如下图,该题属于树和搜索类型的题目,主要考察对于递归方法的使用和树本身遍历方法的理解。本文的题目作者想到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版本的分层遍历方法的速度也还可以,但应该是有更多的方法可以进一步提速的,希望朋友们能够多多指教,非常感谢。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值