Populating Next Right Pointers in Each Node II

题意是说给你一棵二叉树,每个树节点中增加了一个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



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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值