题目描述:
给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
思路一:
把每一层的结点加入到队列,每一层i+1,到下一层时,把上一层在队列中的结点都弹出,按从左到右把下一层的结点逐个加入,如果首次遇到一个结点没有左子结点与右子节点,则返回i
class Solution(object): def minDepth1(self, root): """ 按层次遍历,寻找第一个叶节点(左孩和右孩都为空) """ if not root: return 0 queue = [root] i= 0 while queue: i+=1 size = len(queue) for t in range(size): node = queue.pop(0) if node.left: queue.append(node.left) if node.right: queue.append(node.right) if (not node.left) and not (node.right): return i
思路二:
从根节点开始,如果左右子结点都存在,则把左右子树作为根节点进行递归,选择返回值最小的+1
如果只有左节点存在,则把左节点进行递归,返回值+1
如果只有右节点存在,则把右节点进行递归,返回值+1
class Solution(object): def minDepth(self, root): if root: if root.left and root.right: return 1+min(self.minDepth(root.left),self.minDepth(root.right)) elif root.left: return 1+self.minDepth(root.left) elif root.right: return 1+self.minDepth(root.right) else: return 1 else: return 0
若使用递归求最大深度,把上面代码中的min函数换成max函数即可。
若使用层次遍历求最大深度,去除最后一个if判断,while循环下直接返回i