leetcode104-python 두 갈래 나무 최대 깊이

Given a binary tree, find its maximum depth.
The maximum depth is the number of nodes along the longest path from the root node down to the farthest leaf node.
Note: A leaf is a node with no children.
Example:
Given binary tree  [3,9,20,null,null,15,7] ,
    3
   / \
  9  20
    /  \
   15   7

return its depth = 3.
문제 해결 요점:
1. 하나의 stack을 이용하여 나무에서 흘러나오는 노드를 배치하고 바늘을 설정하여 각 층의 마지막 뒤로 이동하고 층수에 하나를 더한다.
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def maxDepth(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        ans = 0
        if root == None:
            return ans
        st = []
        st.append(root)
        # temp, gp = st[0], st[0]
        gp = st[0]
        ans += 1
        while len(st) != 0:
            temp = st[0]
            if temp.left != None:
                st.append(temp.left)
            if temp.right != None:
                st.append(temp.right)
            
            if gp == temp:
                gp = st[len(st)-1]
                if gp != st[0]:
                    ans += 1
            st.pop(0)
        return ans

또 하나는 귀속적인 방법으로 코드가 비교적 짧기 때문에 앞에 용기를 하나 더 사용할 생각을 하지 않아도 된다.
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def maxDepth(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        
        if root == None:
            return 0
        else:
            return max(self.maxDepth(root.left),self.maxDepth(root.right))+1

좋은 웹페이지 즐겨찾기