파비의 매일매일 공부기록

Today's Challenge 본문

Problem Solving/LeetCode

Today's Challenge

fabichoi 2022. 5. 13. 23:45

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

 

Populating Next Right Pointers in Each Node II - LeetCode

Level up your coding skills and quickly land a job. This is the best place to expand your knowledge and get prepared for your next interview.

leetcode.com

트리는 여전히 어렵다.
동일한 트리 레벨을 표시하도록 구현하면 된다.

class Solution:
    def connect(self, root: 'Node') -> 'Node':
        if not root:
            return None
        
        q = deque()
        q.append(root)
        dummy = Node(-999)
        
        while q:
            length = len(q)
            
            prev = dummy
            
            for _ in range(length):
                popped = q.popleft()
                
                if popped.left:
                    q.append(popped.left)
                    prev.next = popped.left
                    prev = prev.next
                if popped.right:
                    q.append(popped.right)
                    prev.next = popped.right
                    prev = prev.next
        
        return root
반응형

'Problem Solving > LeetCode' 카테고리의 다른 글

Today's Challenge  (0) 2022.05.15
Today's Challenge  (0) 2022.05.14
Today's Challenge  (0) 2022.05.12
Today's Challenge  (0) 2022.05.11
Today's Challenge  (0) 2022.05.10
Comments