LeetCode 노트: 104.Maximum Depth of Binary Tree

질문:

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.

대의:

두 갈래 나무를 주어 가장 큰 깊이를 찾아라.최대 깊이는 뿌리 노드에서 가장 먼 잎 노드까지의 최장 거리의 노드 수를 가리킨다.

생각:

두 갈래 나무의 깊이를 탐색하려면 귀속이 비교적 편리하다.우리 제목이 요구하는 함수는 루트 노드의 깊이를 되돌려줍니다. 그러면 두 갈래 트리의 모든 노드에 이 함수를 호출할 때 루트 노드로 볼 때의 깊이를 되돌려줍니다.예를 들어 모든 잎 노드의 깊이는 1이고 위로 올라가면 2, 3...루트 루트까지 되돌아오는 값이 가장 큰 깊이입니다.모든 노드에 대해 우리는 먼저 그 자체가 노드인지 아닌지를 판단하고 빈 두 갈래 나무라면 0으로 돌아가야 한다.그리고 우리는 두 개의 변수, 하나는 왼쪽 노드의 깊이, 하나는 오른쪽 노드의 깊이를 정의한다.우리는 각각 왼쪽 노드와 오른쪽 노드가 있는지 판단한다. 두 노드 중의 방법은 모두 같다. 왼쪽 노드가 없다고 가정하면 왼쪽 노드의 깊이 변수는 1이고 왼쪽 노드가 있다면 왼쪽 노드의 깊이 변수는 왼쪽 노드가 이 함수를 호출하여 되돌아오는 결과에 1을 더한다.오른쪽 노드에도 같은 조작을 한다.마지막으로 왼쪽 노드의 깊이와 오른쪽 노드의 깊이를 비교하고 누가 큰지 판단하면 어느 변수를 되돌려준다.이렇게 하면 한 층 한 층 귀속되어 최대 깊이를 얻을 수 있다.

코드(Java):

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public int maxDepth(TreeNode root) {
        if (root != null) {//  
            int rightResult;
            int leftResult;
            if (root.left != null) {//  
                leftResult = maxDepth(root.left) + 1;
            } else {//  
                leftResult = 1;
            }
            if (root.right != null) {//  
                rightResult = maxDepth(root.right) + 1;
            } else {//  
                rightResult = 1;
            }
            //  ， 
            return leftResult > rightResult ? leftResult : rightResult;
        } else {//  ， 0
            return 0;
        }
    }
}

그러나 우리가 조금만 더 생각해 보면 코드를 좀 더 간략하게 할 수 있다.우리 코드에서 루트가null인 상황에서 돌아오는 것이 0이기 때문에 사실 왼쪽 노드가 없을 때 정렬은 함수를 사용하여 돌아오는 것도 0이고 1을 더하면 우리가 필요로 하는 1이기 때문에 왼쪽 노드가 있는지 없는지 판단할 필요가 없다. 오른쪽 노드도 마찬가지다.다음과 같이 단순화할 수 있습니다.

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public int maxDepth(TreeNode root) {
        if (root != null) {
            int rightResult = maxDepth(root.left) + 1;
            int leftResult = maxDepth(root.right) + 1;
            return leftResult > rightResult ? leftResult : rightResult;
        } else {
            return 0;
        }
    }
}

함수를 추가로 써서 매개 변수에 깊이를 전달할 수도 있다.

// C++:
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int depth(TreeNode* node, int level) {
        if (node == NULL) return level-1;
        
        int left = depth(node->left, level+1);
        int right = depth(node->right, level+1);
        
        return max(left, right);
    }
    
    int maxDepth(TreeNode* root) {
        return depth(root, 1);
    }
};

집합:https://github.com/Cloudox/LeetCode-Record
작성자 홈페이지 보기