二叉树的最小深度

2020-08-21  本文已影响0人  眼若繁星丶

二叉树的最小深度


LeetCode 111

https://leetcode-cn.com/problems/minimum-depth-of-binary-tree/

解题思路

本题与求二叉树最大深度的思路几乎一样。单独求出左右子树的深度,取两者最小值再加上根节点就是二叉树的最小深度。

值得注意的是,如果样例为[1,2],左子树深度为1,右子树深度为0,则这棵二叉树的最小深度是2,而不是1。右子树为空,不能算上深度为0的子树,即不存在。所以答案应该按照左子树的深度加上根节点即 1 + 1。

考虑到这点,我们就要单独用条件语句来筛选三种情况。

1.左子树空且右子树不空。return 1 + minDepth(root.right);

2.左子树不空且右子树空。return 1 + minDepth(root.left);

3.左子树和右子树都不空。return Math.min(minDepth(root.left), minDepth(root.right)) + 1;

代码如下

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int minDepth(TreeNode root) {
        if (root == null) {
            return 0;
        } 
        int leftDepth = minDepth(root.left);
        int rightDepth = minDepth(root.right);
        if (root.left != null && root.right == null) {
            return 1 + leftDepth;
        } else if (root.left == null && root.right != null) {
            return 1 + rightDepth;
        } else {
            return Math.min(leftDepth, rightDepth) + 1;
        }
    }
}
上一篇下一篇

猜你喜欢

热点阅读