【二叉树算法题记录】最大和最小深度

104.二叉树的最大深度

题目描述

给定一个二叉树root，返回其最大深度。二叉树的 最大深度 是指从根节点到最远叶子节点的最长路径上的节点数。

题目分析

递归法

递归法实际上可以简化成寻找根节点左右子树的最大深度。这样一来我们得到左右子树的最大深度，然后选择较大的那一个深度+1即为整体二叉树的最大深度。不过，我们还得确定遍历的顺序：由于是先得到左右子树的最大深度，再得到根节点的最大深度，所以是左-右-中的顺序，即后序遍历。那么就可以通过递归三部曲写出整体流程：

• 确定函数返回类型与传入参数：int maxDepth(TreeNode* root)
• 确定递归终止条件：如果递归到一个没有孩子的节点node时，即运行到maxDepth(node->left)或maxDepth(node->right)，此时node->left==NULL, node->right==NULL，就要停止往下递归，所以递归条件是：if(root==NULL) return 0;（这里的root就相当于上面的node->left和node->right）
• 确定递归逻辑：也就是往下遍历左子树和右子树的最大深度，然后总结出基于该根节点的最大深度

递归的总体代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int maxDepth(TreeNode* root) {
        // 确定递归终止条件
        if(root==NULL) return 0;
		// 递归逻辑：遍历左右子树，然后计算最大深度
        int left_height = maxDepth(root->left);
        int right_height = maxDepth(root->right);
        int max_height = 1 + max(left_height, right_height);
        return max_height;
    }
};

111.二叉树的最小深度

题目描述

给定一个二叉树，找出其最小深度。最小深度是从根节点到最近叶子节点的最短路径上的节点数量。

题目分析

迭代法

迭代法实际上就是层次遍历，我们遍历到左右孩子都为空的节点就可以返回了，它所在层数一定是二叉树的最小深度。

class Solution {
public:
    int minDepth(TreeNode* root) {
        // 碰到左右孩子都为NULL就返回
        queue<TreeNode*> q;
        if(root!=NULL) q.push(root);
        int layer = 0;  // 二叉树深度
        while(!q.empty()){
            int size = q.size();
            layer++; 
            while(size--){
                TreeNode* node = q.front();
                q.pop();
                if(node->left) q.push(node->left);
                if(node->right) q.push(node->right);
                if(node->left==NULL && node->right==NULL) return layer;
            }
        }
        return layer;
    }
};