이진트리의 높이를 구해보자.
#이진트리를 구성하는 노드
class Node:
def __init__(self,data):
self.right=self.left=None
self.data = data
#이진트리 설계
class Solution:
def insert(self,root,data):
if root==None:
return Node(data)
else:
if data<=root.data:
cur=self.insert(root.left,data)
root.left=cur
else:
cur=self.insert(root.right,data)
root.right=cur
return root
#높이를 구하는 함수
def getHeight(self,root):
#Write your code here
if root is None:
return -1
leftHeight = self.getHeight(root.left) + 1
rightHeight = self.getHeight(root.right) + 1
return max(leftHeight, rightHeight)
T=int(input())
myTree=Solution()
root=None
for i in range(T):
data=int(input())
root=myTree.insert(root,data)
height=myTree.getHeight(root)
print(height)
재귀호출을 사용해서 높이를 구하는 방법이다.
getHeight() 함수가 높이를 구하는 함수이다.
반응형
'IT > [Everyday]Coding' 카테고리의 다른 글
Jewels and Stones (0) | 2019.01.22 |
---|---|
노트북 밧데리 수명 예측 연습 (0) | 2018.02.20 |
이진트리 모든 데이터 탐색 (0) | 2018.02.20 |
컵 모양의 2차 행렬의 모든 합이 가장 큰 것을 알아내라 (0) | 2016.08.16 |
python 선택한 개수 만큼, 일정한 모양으로 출력 연습 (0) | 2016.08.11 |