lintcode 二叉樹的最小深度

1：問題描述：給定一個二叉樹，找出其最小深度。二叉樹的最小深度爲根節點到最近葉子節點的距離。

樣例

給出一棵如下的二叉樹:

        1

     /     \ 

   2       3

          /    \

        4      5  

這個二叉樹的最小深度爲 2

2；解題思路：和求最大深度類似，用遞歸算法，但是比求最大深度複雜，還要考慮斜樹的情況。

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */
class Solution {
public:
    /**
     * @param root: The root of binary tree.
     * @return: An integer
     */
    int minDepth(TreeNode *root) {
        // write your code here
    if(root == NULL) return 0;
    if(root->left == NULL) return minDepth(root->right) + 1;//斜左子樹
    if(root->right == NULL) return minDepth(root->left) + 1;//斜右子樹
 
    int leftDepth = minDepth(root->left);
    int rightDepth = minDepth(root->right);
 
     return leftDepth < rightDepth ? (leftDepth + 1) : (rightDepth + 1);

    }
};

4：解題感想：

剛開始拿到這道題，因爲先做的是求二叉樹的最大深度，求最小深度不就是最大深度的代碼改爲最小就行了嗎？ 這樣做的結果就是有些測試時通不過的。自己想了想終於知道錯哪裏了。因爲如果出現斜樹的時候，只有左子樹，或只有右子樹的時候，如果按照最大深度的那個算法來實現的話，得出的答案是1，明顯是錯誤的。判斷左子樹或右子樹是否爲空，若左子樹爲空，則返回右子樹的深度，反之返回左子樹的深度，如果都不爲空，則返回左子樹和右子樹深度的最小值+1.