leetcode @python 116. Populating Next Right Pointers in Each Node

最新推荐文章于 2025-09-11 19:19:55 发布

weixin_30824479

最新推荐文章于 2025-09-11 19:19:55 发布

阅读量67

点赞数

CC 4.0 BY-SA版权

文章标签：数据结构与算法 python

原文链接：http://www.cnblogs.com/slurm/p/5318710.html

本文详细解析了如何使用递归方法在二叉树的每层节点间建立链表连接，包括核心解题思路与代码实现，旨在帮助读者深入理解二叉树结构与链表操作。

题目链接

https://leetcode.com/problems/populating-next-right-pointers-in-each-node/

题目原文

题目大意

二叉树的每个节点都包含左子树和右子树，将同一层的子树用链表连接起来

解题思路

递归求解：
1.如果该节点有左子树：1）该节点有右子树 2）该节点没有右子树递归该节点的左子树
2.如果该节点有右子树：1）该节点的next节点有左子树 2）该节点的next节点没有左子树

代码

class TreeLinkNode(object):
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None
        self.next = None


class Solution(object):
    def connect(self, root):
        """
        :type root: TreeLinkNode
        :rtype: nothing
        """
        if not root:
            return
        else:
            root.next = None
            self.help(root)

    def help(self, node):
        if node.left:
            if node.right:
                node.left.next = node.right
            else:
                node.left.next = None
            self.help(node.left)
        if node.right:
            if node.next and node.next.left:
                node.right.next = node.next.left
            else:
                node.right.next = None
            self.help(node.right)

转载于:https://www.cnblogs.com/slurm/p/5318710.html