LeetCode——111 二叉树的最小深度(JAVA)

给定一个二叉树，找出其最小深度。

最小深度是从根节点到最近叶子节点的最短路径上的节点数量。

说明： 叶子节点是指没有子节点的节点。

示例 1：

输入：root = [3,9,20,null,null,15,7]
输出：2

示例 2：

输入：root = [2,null,3,null,4,null,5,null,6]
输出：5

提示：

• 树中节点数的范围在 [0, 10⁵] 内
• -1000 <= Node.val<= 1000

思路

思路和之前的二叉树遍历差不多一致，主要就是到叶子结点之后，判断一下当前arrayList里的元素数目，此时就是该路径对应的深度，如果小于事先定义的minDeep，则更新即可。

这里要注意的一点是，最好在DFS时做好剪枝，不剪枝时虽然也能AC，但是用时20ms，剪完直接2ms。

具体的话就是在每次将要进行下一层DFS时先判断一下：如果当前所在结点的深度已经大于或等于minDeep了，那也就无需再往下一层走了（往下一层走深度必然会增加，但是题目只要求返回最小深度，因此没必要浪费时间搜索下去）。

代码

public class Solution {
	public static class TreeNode {//为了方便本地调试调用内部类,我写了静态的
		//实际提交时可以不写TreeNode类
		int val;
		TreeNode left;
		TreeNode right;
		TreeNode() {}
		TreeNode(int val) { this.val = val; }
		TreeNode(int val, TreeNode left, TreeNode right) {
			this.val = val;
			this.left = left;
			this.right = right;
		}
	}
	int minDeep = 123123123;
	ArrayList<Integer> arrayList = new ArrayList<Integer>();
	public void dfs(TreeNode root) {
		arrayList.add(root.val);
		if(root.left==null&&root.right==null) {
			if(arrayList.size()<minDeep) minDeep = arrayList.size();
		}
		if(root.left!=null&&arrayList.size()<minDeep) {
			dfs(root.left);
			arrayList.remove(arrayList.size()-1);
		}
		if(root.right!=null&&arrayList.size()<minDeep) {
			dfs(root.right);
			arrayList.remove(arrayList.size()-1);
		}
	}
	public int minDepth(TreeNode root) {
		if(root==null) {
			return 0;
		}
		else {
			dfs(root);
			return minDeep;
		}
	}
}