利用python 完成leetcode 116 填充每个节点的下一个右侧节点指针

最新推荐文章于 2022-01-14 21:34:57 发布

原创最新推荐文章于 2022-01-14 21:34:57 发布 · 515 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #python #算法 #二叉树

leetcode 同时被 2 个专栏收录

137 篇文章

订阅专栏

leetcode 中等

67 篇文章

订阅专栏

本文介绍了如何利用Python解决LeetCode中的116题，即填充每个节点的下一个右侧节点指针。通过对完美二叉树的特性分析，提出了解决思路：当左节点存在时，将右节点的next指针指向左节点；当当前节点有右侧节点时，将其next指针指向其父节点的下一个左侧节点；否则，next指针设为None。并给出了具体的代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每个 next 指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL。

初始状态下，所有 next 指针都被设置为 NULL。

示例：

在这里插入图片描述

输入：{“ $KaTeX parse error: Expected '}', got 'EOF' at end of input: \dots":"1","left":{"$ id”:“2”,“left”:{“ $KaTeX parse error: Expected 'EOF', got '}' at position 53: \dotst":null,"val":4}̲,"next":null,"r\dots$ id”:“4”,“left”:null,“next”:null,“right”:null,“val”:5},“val”:2},“next”:null,“right”:{“ $KaTeX parse error: Expected '}', got 'EOF' at end of input: \dots":"5","left":{"$ id”:“6”,“left”:null,“next”:null,“right”:null,“val”:6},“next”:null,“right”:{"$id":“7”,“left”:null,“next”:null,“right”:null,“val”:7},“val”:3},“val”:1}

输出：{“ $KaTeX parse error: Expected '}', got 'EOF' at end of input: \dots":"1","left":{"$ id”:“2”,“left”:{“ $KaTeX parse error: Expected '}', got 'EOF' at end of input: \dots:null,"next":{"$ id”:“4”,“left”:null,“next”:{“ $KaTeX parse error: Expected '}', got 'EOF' at end of input: \dots:null,"next":{"$ id”:“6”,“left”:null,“next”:null,“right”:null,“val”:7},“right”:null,“val”:6},“right”:null,“val”:5},“right”:null,“val”:4},“next”:{“ $KaTeX parse error: Expected '}', got 'EOF' at end of input: \dots":"7","left":{"$ ref”:“5”},“next”:null,“right”:{“ $KaTeX parse error: Expected 'EOF', got '}' at position 9: ref":"6"}̲,"val":3},"righ\dots$ ref”:“4”},“val”:2},“next”:null,“right”:{"$ref":“7”},“val”:1}

解释：给定二叉树如图 A 所示，你的函数应该填充它的每个 next 指针，以指向其下一个右侧节点，如图 B 所示。
思路
因为为完美二叉树，所以当左节点不为空时，右节点也不为空，root.left.next=root.right(示例的2，4，6节点)
当root有右侧节点，root.right.next=root.next.left(示例的5节点)
否则root.right.next=None(示例的1,3,7节点)

代码

def connect(self, root: 'Node') -> 'Node':
    
        if(root==None):return root
        if(root.left==None):return root
        root.left.next=root.right
        if root.next!=None:root.right.next=root.next.left
        else:root.right.next=None
        self.connect(root.left)
        self.connect(root.right)
        return root