LeetCode【111. 二叉树的最小深度】

最新推荐文章于 2024-04-29 11:29:27 发布

转载最新推荐文章于 2024-04-29 11:29:27 发布 · 78 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/wzwi/p/10768417.html

文章标签：

#数据结构与算法

本文探讨了二叉树最小深度算法的实现与修正，针对原有算法在处理特定情况时的不足进行了改进，确保了算法能正确计算到叶子节点而非仅限于根节点，提供了一种更精确的递归解决方案。

最小深度，看起来很简单，就是左右节点的深度最小值

定义一个函数，计算其深度

class Solution {
    public int minDepth(TreeNode root) {
        if(root == null)
        {
            return 0;
        }
        else if(root.left == null && root.right ==null)
        {
            return 1;
        }
        else
        {
            int left = Depth(root.left);
            int right = Depth(root.right);
            if(left > right)
            {
            return left;
            }
            else
            {
                return right;
            }
        }
    }
    public int Depth(TreeNode root)
    {
        if(root == null)
        {
            return 0;
        }
        else
        {
            int left = Depth(root.left);
            int right = Depth(root.right);
            if(left<right)
            {
                return left+1;
            }
            else
            {
                return right+1;
            }
        }
    }
}

有错误在于，[1,2]，只有一个节点，然后，该代码就会直接输出1，因为另一个没有节点，那么就直接是1

但实际上，应该是到叶子节点，不应该有叶子节点，而仍然只算根节点。

而且定义的函数其实和本函数类似，那么就可以直接迭代本函数，不需要另外定义。

class Solution {
    public int minDepth(TreeNode root) {
        if(root == null)
        {
            return 0;
        }
        else if(root.left == null && root.right ==null)
        {
            return 1;
        }
        else
        {
            int left = minDepth(root.left);
            int right = minDepth(root.right);
            if(left == 0 && right > 0)
            {
                return right+1;
            }
            else if(right == 0 && left > 0)
            {
                return left+1;
            }
            else if(left > right)
            {
                return right+1;
            }
            else
            {
                return left+1;
            }
        }
    }
}