[LeetCode] 2236번: 트리 구조

Goal

1. 트리 구조 이해하기
2. 가장 간단한 트리 구조 알고리즘 문제 풀어보기

 

트리 구조란?

트리 구조란 노드(Node)와 자식노드(Child Node)로 구성된 구조를 얘기한다.

가장 흔히 보는 트리 구조는 바이너리 트리 구조(Binary Tree)로 각 노드가 왼쪽 오른쪽으로 자식노드 2개를 가진 형태다.

간단한 트리 구조

 

LeetCode 2236번: Root Equals Sum of Children (풀어보기)

Hint

1. root라는 트리 구조의 정의 :
      root.val = 현재 노드의 수
      root.left = 현재 노드의 왼쪽 자식노드
      root.right= 현재 노드의 오른쪽 자식노드
2. root.val이 현재 노드의 수라면 root.left 노드의 수를 구하려면? (정답은 밑에 하이라이트하면 보임)
      root.left.val = 현재 노드의 왼쪽 자식노드의 수
      root.right.val = 현재 노드의 오른쪽 자식노드의 수

 

Solution

class Solution:
    def checkTree(self, root: Optional[TreeNode]) -> bool:
    	# 현재 노드의 왼쪽 자식노드의 수와 오른쪽 자식노드의 수를 더한 후 비교
        if root.val == root.left.val + root.right.val: 
            return True
        return False