Leetcode 116. 각 노드의 다음 오른쪽 노드 포인터 채우기

3724 단어 LeetCode
Time: 20190901 Type: Medium

제목 설명


완벽한 두 갈래 나무를 정하는데 그 모든 잎 노드는 같은 층에 있고 각 부 노드는 두 개의 자 노드가 있다.두 갈래 트리는 다음과 같이 정의됩니다.
struct Node {
  int val;
  Node *left;
  Node *right;
  Node *next;
}

next 바늘을 채워서 다음 오른쪽 노드를 가리키도록 합니다.다음 오른쪽 노드를 찾지 못하면 next 포인터를 NULL 로 설정합니다.
초기 상태에서는 모든 포인터nextNULL로 설정됩니다.
출처: 리코드(LeetCode) 링크:https://leetcode-cn.com/problems/populating-next-right-pointers-in-each-node저작권은 인터넷 소유에 귀속된다.상업 전재는 정부에 연락하여 권한을 부여하고, 비상업 전재는 출처를 명시해 주십시오.

생각


두 갈래 나무가 가득 차 있기 때문에 귀속적인 사고방식으로 풀 수 있다. 왼쪽 아이의next지향은 오른쪽 아이, 오른쪽 아이의지향은 아버지 노드의next지향은 왼쪽 아이이다.귀속의 성질을 만족시키다.117과는 달리 117과 본제는 모두 대열로 풀 수 있다.

코드

'''
# Definition for a Node.

class Node:
    def __init__(self, val, left, right, next):
        self.val = val
        self.left = left
        self.right = right
        self.next = next
'''

class Solution:
    def connect(self, root: 'Node') -> 'Node':
        if not root:
            return 
        #  next 
        if root.left:
            root.left.next = root.right
            
        if root.right and root.next:
            root.right.next = root.next.left
            
        self.connect(root.left)
        self.connect(root.right)
        
        return root

END.

좋은 웹페이지 즐겨찾기