力扣题解-111. 二叉树的最小深度(分治法思想-DFS方法解决)

题目: 111. 二叉树的最小深度

给定一个二叉树，找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明: 叶子节点是指没有子节点的节点。
示例:
给定二叉树 [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回它的最小深度 2.

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/minimum-depth-of-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解

之前做过求二叉树的最大深度的问题力扣题解-104. 二叉树的最大深度, 实际上求二叉树的最小深度与最大深度十分相似.

这里再次重复一下如何利用Divide-and-Conquer的思想分析这类问题. 分治法的思想在文章Divide and Conquer思想及实际应用中有介绍.

分治法的思想是将原问题拆分成更小的子问题, 然后对子问题进行递归的拆分,直到子问题不能拆分为止, 最后将子问题的解合并得到原问题的解.

具体来说, 将原二叉树分解为根节点、左子树和右子树, 通过计算两个子树的最小深度, 以获取原树的最小深度, 即可得到原问题的解.

我们直接套用分治法的步骤来寻找解决问题的方法.

分治法步骤

按照分治法的思想, 分成三步来解决问题:

1. devide: 将原二叉树分解为根节点、左二叉树和右二叉树.
2. conquer: 递归地求解左二叉树的最小深度和右二叉树的最小深度.
3. combine: 原始二叉树的最小深度等于左子树最小深度与右子树最小深度的最小值 + 1.

隐藏坑

    1
   /
  2
 /
3

我们看一下这个二叉树. 该二叉树只有左子树, 右子树为空.

我们来看一下, 本题中最小深度的定义: 最小深度是从根节点到最近叶子节点的最短路径上的节点数量。

也就是说, 当根节点的右子树为空时, 二叉树的最小深度就是左子树的最小深度了. 这里后面代码能够看到处理.

DFS实现

/**
 * 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 minDepth(TreeNode* root) {
        if (root == nullptr) {
            return 0;
        }

        if (root->left == nullptr) {
            return 1 + minDepth(root->right);
        }
        if (root->right == nullptr) {
            return 1 + minDepth(root->left);
        }
        return 1 + min(minDepth(root->left), minDepth(root->right));
    }
};