Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

Python에서 가장 깊은 잎의 가장 낮은 공통 조상


루트된 이진 트리가 있다고 가정하면 가장 깊은 잎의 가장 낮은 공통 조상을 반환해야 합니다. 우리는 다음을 명심해야 합니다 -

  • 이진 트리의 노드는 자식이 없는 경우에만 리프 노드입니다.

  • 트리의 루트 깊이는 0이고 노드의 깊이가 d일 때 각 자식의 깊이는 d+1입니다.

  • S의 모든 노드가 루트 A가 있는 하위 트리에 있도록 최대 깊이를 갖는 노드 A의 노드 집합 S의 가장 낮은 공통 조상입니다.

입력이 [1,2,3,4,5]인 경우

Python에서 가장 깊은 잎의 가장 낮은 공통 조상

그러면 출력은 [2,4,5]

가 됩니다.

이 문제를 해결하기 위해 다음 단계를 따릅니다. −

  • solve()라는 메서드를 정의하면 노드가 사용되며 다음과 같이 작동합니다. -

  • 노드가 없으면 [0, 없음]

    으로 목록을 반환합니다.
  • 왼쪽 및 오른쪽 하위 트리에 노드가 없으면 [1, 없음]

    이 있는 목록을 반환합니다.
  • d1, l :=solve(노드의 왼쪽), d2, r :=solve(노드의 오른쪽)

  • d1> d2이면 값이 [d1 + 1, l]

    인 목록을 반환합니다.
  • 그렇지 않으면 d2> d1일 때 값이 [d2 + 1, r]

    인 목록을 반환합니다.
  • 값이 [d1 + 1, node]

    인 목록을 반환합니다.
  • 주요 방법에서는 다음을 수행합니다 -

  • 목록 :=해결(루트)

  • 반환 목록[1]

예제(파이썬)

더 나은 이해를 위해 다음 구현을 살펴보겠습니다. −

class TreeNode:
   def __init__(self, data, left = None, right = None):
      self.data = data
      self.left = left
      self.right = right
def insert(temp,data):
   que = []
   que.append(temp)
   while (len(que)):
      temp = que[0]
      que.pop(0)
      if (not temp.left):
         if data is not None:
            temp.left = TreeNode(data)
         else:
            temp.left = TreeNode(0)
         break
      else:
         que.append(temp.left)
      if (not temp.right):
         if data is not None:
            temp.right = TreeNode(data)
         else:
            temp.right = TreeNode(0)
         break
      else:
         que.append(temp.right)
def make_tree(elements):
   Tree = TreeNode(elements[0])
   for element in elements[1:]:
      insert(Tree, element)
   return Tree
def print_tree(root):
   #print using inorder traversal
   if root is not None:
      print_tree(root.left)
      print(root.data, end = ', ')
      print_tree(root.right)
class Solution(object):
   def lcaDeepestLeaves(self, root):
      return self.solve(root)[1]
   def solve(self,node):
      if not node:
         return [0,None]
      if not node.left and not node.right:
         return [1,node]
      d1,l = self.solve(node.left)
      d2,r = self.solve(node.right)
      if d1>d2:
         return [d1+1,l]
      elif d2>d1:
         return [d2+1,r]
      return [d1+1,node]
ob = Solution()
root = make_tree([1,2,3,4,5])
print_tree(ob.lcaDeepestLeaves(root))

입력

[1,2,3,4,5]

출력

4, 2, 5,