Leetcode做题日记：117. 填充每个节点的下一个右侧节点指针 II(PYTHON)

最新推荐文章于 2024-05-29 05:50:00 发布

i火柴天堂

最新推荐文章于 2024-05-29 05:50:00 发布

阅读量368

点赞数

分类专栏： leetcode 文章标签： Leetcode

本文链接：https://blog.csdn.net/weixin_44033136/article/details/90718180

版权

leetcode 专栏收录该内容

113 篇文章 0 订阅

订阅专栏

给定一个二叉树

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: \dotsnull,"right":{"$ id”:“6”,“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 'EOF', got '}' at position 53: \dotst":null,"val":7}̲,"right":null,"\dots$ id”:“6”,“left”:null,“next”:null,“right”:{“ $KaTeX parse error: Expected 'EOF', got '}' at position 9: ref":"5"}̲,"val":3},"righ\dots$ ref”:“4”},“val”:2},“next”:null,“right”:{"$ref":“6”},“val”:1}

解释：给定二叉树如图 A 所示，你的函数应该填充它的每个 next 指针，以指向其下一个右侧节点，如图 B 所示。

提示：

你只能使用常量级额外空间。
使用递归解题也符合要求，本题中递归程序占用的栈空间不算做额外的空间复杂度。

第一次的代码：
这题和上一题的区别是，这题不是完美二叉树。即叶子可能不在同一层。

		def cn(r):
            if not r:
                return
            i=0
            j=1
            while j<len(r):#将该层连接起来
                r[i].next=r[j]
                i=i+1
                j=j+1
            nr=[]
            for i in r:#将下一层储存起来
                if i.left:
                    nr.append(i.left)
                if i.right:
                    nr.append(i.right)
            cn(nr)
            return r[0]
        if not root:
            return root
        return cn([root])

232ms，排名42%

当然，递归方式有很多
这里注意先递归右树，因为递归的时候需要用到根节点的.next节点，若右树没有处理，则无法完成递归，而恰好右树的next为空。

		def cn(r):
            if not r:
                return
            if r.left and r.right:#完成本节点的next
                r.left.next=r.right
            p=r.right if r.right else r.left#选择next到右树
            p2=r.next#选择右树根节点
            while p and p2:#完成next
                if p2.left:
                    p.next=p2.left
                    break
                elif p2.right:
                    p.next=p2.right
                    break
                else:#右树无叶节点，再选next根节点为右树
                    p2=p2.next
            cn(r.right)
            cn(r.left)
        if not root:return root
        cn(root)
        return root