1. 程式人生 > >Leetcode:111.二叉樹的最小深度

Leetcode:111.二叉樹的最小深度

給定一個二叉樹,找出其最小深度。

最小深度是從根節點到最近葉子節點的最短路徑上的節點數量。

說明: 葉子節點是指沒有子節點的節點。

示例:

給定二叉樹 [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回它的最小深度  2.

解題思路:

遞迴。從根節點開始,尋找他的左子樹的深度depth_left,以及右子樹的深度depth_right,和然後返回min(depth_left,depth_right)+1。與找最大不同是,如果當前層的層數大於最小值,則無需訪問該節點。

                                                         

C++程式碼
#define hasLChild(x) (!(x->left==NULL))
#define hasRChild(x) (!(x->right==NULL))
#define hasNoChild(x) (!hasLChild(x)&&!hasRChild(x))
class Solution {
public:
    int minDepth(TreeNode* root) {
        if (root == NULL) return 0;
        DFS(root,1);
        return min_depth;
    }
    void DFS(TreeNode* root,int depth) {
        if (depth >= min_depth) return;
        if (hasNoChild(root)) { min_depth = (depth < min_depth ? depth : min_depth); return; }
        if (hasLChild(root)) { DFS(root->left, depth + 1); }
        if (hasRChild(root)) { DFS(root->right, depth + 1); }
    }
public:
    int min_depth = INT16_MAX;
};