【LeetCode】117. Populating Next Right Pointers in Each Node II 解題報告(Python)

給定一個二叉樹

struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每個 next 指針,讓這個指針指向其下一個右側節點。如果找不到下一個右側節點,則將 next 指針設置爲 NULL。

初始狀態下,所有 next 指針都被設置爲 NULL。

進階:

你只能使用常量級額外空間。
使用遞歸解題也符合要求,本題中遞歸程序佔用的棧空間不算做額外的空間複雜度。

在這裏插入圖片描述
解題思路1:

這一題和【LeetCode】116. Populating Next Right Pointers in Each Node很類似,完全可以一樣採用隊列層序遍歷的方法(指向下一個會有一個判斷,不能時當前層的最後一個,及1,3,7,他們都直接指向空)

1進隊列,1出隊列,1指向隊列下一個空,1左(2)進隊,1右(3)進隊
2出隊列,2指向隊列下一個3,2左(4)進隊,2右(5)進隊
3出隊列,3指向隊列下一個空,3右(7)進隊
4出隊列,4指向隊列下一個5,3右(7)進隊
隊列有457,然後,逐個出隊列並指向下一個,最後7指向空,就完事了

提交代碼1:(隊列層序,Runtime: 52 ms, faster than 64.76 % )

class Solution:
    def connect(self, root: 'Node') -> 'Node':
        if not root:
            return root
        Q = collections.deque([root])
        while Q:
            size = len(Q)
            for i in range(size):
                node = Q.popleft()
                if i < size - 1:
                    node.next = Q[0]
                if node.left:
                    Q.append(node.left)
                if node.right:
                    Q.append(node.right)
        return root

解題思路2:

【LeetCode】116. Populating Next Right Pointers in Each Node思路二類似,請參考這個,https://leetcode-cn.com/problems/populating-next-right-pointers-in-each-node-ii/solution/tian-chong-mei-ge-jie-dian-de-xia-yi-ge-you-ce-j-4/

提交代碼2:(Runtime: 52 ms, faster than 64.76 % )

class Solution:

    def processChild(self, childNode, prev, leftmost):
        if childNode:
            if prev:
                prev.next = childNode
            else:
                leftmost = childNode
            prev = childNode
        return prev, leftmost

    def connect(self, root: 'Node') -> 'Node':
        if not root:
            return root
        leftmost = root
        while leftmost:
            prev, curr = None, leftmost
            leftmost = None
            while curr:
                prev, leftmost = self.processChild(curr.left, prev, leftmost)
                prev, leftmost = self.processChild(curr.right, prev, leftmost)
                curr = curr.next

        return root
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章