python leetcode 116. Populating Next Right Pointers in Each Node 117 II

最新推荐文章于 2022-02-28 18:16:11 发布

Neekity

最新推荐文章于 2022-02-28 18:16:11 发布

阅读量685

点赞数

CC 4.0 BY-SA版权

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

本文链接：https://blog.youkuaiyun.com/Neekity/article/details/84974427

python 同时被 2 个专栏收录

263 篇文章

订阅专栏

leetcode

221 篇文章

订阅专栏

本文深入探讨了两个二叉树节点链接算法的实现，分别适用于完全二叉树和一般的二叉树。通过栈结构迭代遍历树的节点，巧妙地将同一层的相邻节点进行链接，从而实现了Populating Next Right Pointers in Each Node 和 Populating Next Right Pointers in Each Node II 的解决方案。

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

不难，考察代码编写能力

116. Populating Next Right Pointers in Each Node

class Solution:
    # @param root, a tree link node
    # @return nothing
    def connect(self, root):
        if not root:
            return None 
        def isP(n):
            for i in range(1,n+1):
                if 2**i-1==n:
                    return True 
                elif 2**i>n:
                    return False
            return False
        def isQ(n):
            for i in range(1,n):
                if 2**i==n:
                    return True 
                elif 2**i>n:
                    return False
            return False
        stack=[root]
        count=0
        pre=TreeLinkNode(0)
        while stack:
            now = stack.pop(0)
            count+=1
            if isQ(count):
                pre=now 
            else:
                pre.next=now
                pre=pre.next
            if now.left:
                stack.append(now.left)
                stack.append(now.right)

117. Populating Next Right Pointers in Each Node II

class Solution:
    # @param root, a tree link node
    # @return nothing
    def connect(self, root):
        if not root:
            return None 
    
        stack=[root]
        
        while stack:
            mystack=[]
            pre=None
            while stack:
                curr=stack.pop()
                if pre:
                    pre.next=curr 
                pre=curr
                mystack.append(curr)
            while mystack:
                node = mystack.pop()
                if node.right:
                    stack.append(node.right)
                if node.left:
                    stack.append(node.left)