给定一个二叉树、找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。如图案例
算法思路:
遍历整颗树,找到每一个叶子节点,从叶子节点往上开始计算,
左右子节点都为空则记录深度为1。
左右子节点只有一边,深度记录为子节点深度+1 。
左右两边都有子节点,则记录左右子节点的深度较小值+1。
算法实现代码
package com.hx;
import java.util.LinkedList;
import java.util.Queue;
public class TreeDeep {
static class TreeNode {
private int val;
private TreeNode left;
private TreeNode right;
public TreeNode(int val, TreeNode left, TreeNode right) {
this.val = val;
this.left = left;
this.right = right;
}
}
static class QueueNode {
private TreeNode node;
private int deep;
public QueueNode(TreeNode node, int deep) {
this.deep = deep;
this.node = node;
}
}
public static void main(String[] args) {
TreeNode node7 = new TreeNode(7,null,null);
TreeNode node6 = new TreeNode(6,node7,null);
TreeNode node5 = new TreeNode(5,null,null);
TreeNode node4 = new TreeNode(4,null,null);
TreeNode node3 = new TreeNode(3,node6,null);
TreeNode node2 = new TreeNode(2,node4,node5);
TreeNode node1 = new TreeNode(1,node2,node3);
System.out.println(minDeep(node1));
}
/**
* 求最小深度
* @param root
* @return
*/
public static int minDeep(TreeNode root) {
if(null == root) {
return 0;
}
//判断左右两边叶子节点为空,是返回深度1
if(null == root.left && null == root.right) {
return 1;
}
//定义最小深度变量
int min = Integer.MAX_VALUE;
//
if(root.left != null) {
min = Math.min(minDeep(root.left),min);
}
if(root.right != null) {
min = Math.min(minDeep(root.right),min);
}
return min+1;
}
}
执行结果
时间复杂度:O(N)
空间复杂度:O(logN) 取决于树的高度