二叉树的深度

二叉树的最小深度

描述

给定一个二叉树，找出其最小深度。

最小深度是从根节点到最近叶子节点的最短路径上的节点数量。

说明: 叶子节点是指没有子节点的节点。

示例:

给定二叉树 [3,9,20,null,null,15,7],

3

/
9 20
/
15 7
返回它的最小深度 2.

思路和代码

递归：

• 如果root == NULL，返回0
• 如果root->leftNULL && root->rightNULL，是叶结点返回1；
• 如果左右子树有一个为空，返回的应该是另一个子树的深度+1
• 如果左右子数都不为空，返回的是左右子树深度较小值加1

class Solution {
public:
    int minDepth(TreeNode* root) {
        if(root == NULL)
            return 0;
        if(root->left == NULL && root->right == NULL)
            return 1;
        if(root->left == NULL)
            return minDepth(root->right) + 1;
        if(root->right == NULL)
            return minDepth(root->left) + 1;
        return min(minDepth(root->left), minDepth(root->right)) + 1;
    }
};

迭代：

class Solution {
public:
    int minDepth(TreeNode* root) {
        if(root == NULL)
            return 0;
        int res = INT_MAX;
        stack<pair<TreeNode*, int>> s;
        s.push(make_pair(root, 1));
        while(!s.empty())
        {
            auto node = s.top().first;
            int depth = s.top().second;
            s.pop();

            //如果node是叶子结点，则更新res
            if(node->left == NULL && node->right == NULL)
                res = min(res, depth);

            //左子树不为空
            if(node->left && res > depth)//剪枝，如果目前的深度小于res,则继续，如果目前的深度已经大于已知的叶结点最小值，就不用再往下遍历，因为不会有更小的叶结点
                s.push(make_pair(node->left, depth + 1));
            if(node->right && res > depth)
                s.push(make_pair(node->right, depth + 1));
        }
        return res;
    }
};

二叉树的最大深度

不同于上一题，这一题求得是最大深度

递归：

class Solution {
public:
    int maxDepth(TreeNode* root) {
        if(root == NULL)
            return 0;
        int max_left=maxDepth(root->left);
        int max_right=maxDepth(root->right);
        return max_left>max_right ? max_left+1 : max_right+1;
    }
};

迭代：可以仿照上面。