111. 二叉树的最小深度

https://leetcode-cn.com/problems/minimum-depth-of-binary-tree/
思路1:dfs递归

class Solution:
    def minDepth(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """        
        #递归出口,空节点
        if not root:
            return 0
        
        min_depth = sys.maxsize
        #dfs  
        min_depth = min(self.minDepth(root.left), min_depth)        
        min_depth = min(self.minDepth(root.right), min_depth)
        
        return min_depth + 1

误区:对[1,2]这组数据,输出是1,但是正确为2,因为右子为空,不构成叶子。
需要对叶节点额外考虑

class Solution:
    def minDepth(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """        
        if not root:
            return 0
        #递归出口,没有左右子,叶节点
        if not root.left and not root.right:
            return 1
        
        min_depth = sys.maxsize  #最小深度假设为最大整数
        #若左子存在,对其dfs递归
        if root.left:   
            min_depth = min(self.minDepth(root.left), min_depth)
        #若右子存在,对其dfs递归
        if root.right:
            min_depth = min(self.minDepth(root.right), min_depth)
        
        return min_depth + 1
        

思路2:层序。从根开始计算深度,初始=1。遇到叶节点就查看当前深度,并与最小深度记录比较。

import queue
class Solution:
    def minDepth(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        if not root:
            return 0
        depth = 0
        min_depth = sys.maxsize  #记录最小深度
        q = queue.Queue()
        q.put(root)
        #层序
        while not q.empty():
            depth += 1
            count = q.qsize()
            for _ in range(count):
                tmp = q.get()
                if tmp.left:
                    q.put(tmp.left)
                if tmp.right:
                    q.put(tmp.right)
                if not tmp.left and not tmp.right:
                    min_depth = min(depth, min_depth)
        
        return min_depth
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值