짱아의 개발 기록장

LeetCode : 104. Maximum Depth of Binary Tree 본문

Algorithm/LeetCode

LeetCode : 104. Maximum Depth of Binary Tree

jungahshin 2021. 1. 6. 11:22
반응형

정말 간단한 dfs로 트리의 높이를 구하는 문제였다.

머리를 깨워주기 위해..ㅎㅎ

 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
class Solution {
public:
    int height = 1;
    
    void countMaxHeight(TreeNode* node, int h){
        if(node==NULL){
            height = max(height, h);
            return;
        }
        
        countMaxHeight(node->left, h+1);
        countMaxHeight(node->right, h+1);
    }
    
    int maxDepth(TreeNode* root) {
        if(root==NULLreturn 0;
        countMaxHeight(root->left, 1);
        countMaxHeight(root->right, 1);
        
        return height;
    }
};
cs
반응형
Comments