[HackerRank] Tree: Height of a Binary Tree

https://www.hackerrank.com/challenges/tree-height-of-a-binary-tree/problem

Tree: Height of a Binary Tree | HackerRank

Data Structures > Trees

 

이진 트리의 높이를 구하는 함수 getHeight() 를 작성한다.

 

 

코드

int getHeight(struct node* root) {
    int leftH = 0, rightH = 0;
    if (root->left) 
        leftH = 1 + getHeight(root->left);
    if (root->right)
        rightH = 1 + getHeight(root->right);
    return leftH > rightH ? leftH : rightH;
}

 

설명

getHeight 함수를 재귀 호출해 root 노드를 기준으로 왼쪽의 높이를 leftH, 오른쪽의 높이를 rightH에 저장한다.

leftH와 rightH 중 큰 것이 전체 트리의 높이이므로 해당 값을 리턴한다.