leetcode - Tree - Maximum Depth of Binary Tree

이번 문제는 주어진 트리의 max depth를 찾는 문제입니다.

 

완전탐색을 통해 트리의 모든 노드를 들리며 depth의 max를 구했습니다.

 

코드는 아래와 같습니다.

 

class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

class Solution:
    def maxDepth(self, root: TreeNode) -> int:

        if root == None: return 0
        if root.right == None and root.left == None: return 1
        depth = 1
        depth = max(self.find(root.right, depth), self.find(root.left, depth))
        return depth

    def find(self, root: TreeNode, depth: int):
        if root == None: return depth
        return max(self.find(root.right, depth+1), self.find(root.left, depth+1))

 

감사합니다.