[Leetcode][python]Populating Next Right Pointers in Each Node I and II/填充同一层的兄弟节点

【Leetcode题解】 同时被 2 个专栏收录
146 篇文章 0 订阅

Populating Next Right Pointers in Each Node

题目大意

为二叉树的节点都添加一个next指针,指向跟它在同一高度的右边的节点,如果右边没有节点,就指向None。

解题思路

递归,容易看懂。

代码

# Definition for binary tree with next pointer.
# class TreeLinkNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
#         self.next = None

class Solution:
    # @param root, a tree link node
    # @return nothing
    def connect(self, root):
        if root and root.left:  # 如有root和左孩子
            root.left.next = root.right  # 左孩子指向右孩子
            if root.next:  # 如链表有右边,那么其右孩子指向右边节点左孩子
                root.right.next = root.next.left
            else:
                root.right.next = None
            self.connect(root.left)
            self.connect(root.right)

Populating Next Right Pointers in Each Node II

题目大意

二叉树并不都是满二叉树

解题思路

还是递归,但是情况考虑清楚
参考:http://www.cnblogs.com/zuoyuan/p/3745369.html

代码

class Solution:
    # @param root, a tree link node
    # @return nothing
    def connect(self, root):
        if root:
            if root.left and root.right:
                root.left.next = root.right
                tmp = root.next
                while tmp:  # 直到右边有第一个左孩子或者右孩子,连上后就break
                    if tmp.left: root.right.next = tmp.left; break
                    if tmp.right: root.right.next = tmp.right; break
                    tmp = tmp.next
            elif root.left:
                tmp = root.next  # 即使有null,下面不遍历而已,并不会报错
                while tmp:  # 用左孩子去连接
                    if tmp.left: root.left.next = tmp.left; break
                    if tmp.right: root.left.next = tmp.right; break
                    tmp = tmp.next
            elif root.right:
                tmp = root.next
                while tmp:  # 用右孩子去连接
                    if tmp.left: root.right.next = tmp.left; break
                    if tmp.right: root.right.next = tmp.right; break
                    tmp = tmp.next
            # 必须先遍历右边子树
            self.connect(root.right)
            self.connect(root.left)

总结

两道题目都是链表和二叉树结合,其实不难。
也还有很多解法,这两个比较通俗易懂。

  • 0
    点赞
  • 0
    评论
  • 0
    收藏
  • 一键三连
    一键三连
  • 扫一扫,分享海报

©️2021 CSDN 皮肤主题: Age of Ai 设计师:meimeiellie 返回首页
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值