leetcode 111. 二叉树的最小深度
一、题目
给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明:叶子节点是指没有子节点的节点。
输入:root = [2,null,3,null,4,null,5,null,6]
输出:5
二、解法
- 递归:
/**
* Definition for a binary tree node.
* public class 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;
* }
* }
*/
class Solution {
public int minDepth(TreeNode root) {
if(root==null) return 0;
if(root.left==null) return 1+minDepth(root.right);
else if(root.right==null) return 1+minDepth(root.left);
else return 1+Math.min(minDepth(root.left),minDepth(root.right));
}
}
- 迭代:
用bfs广度优先搜索:搜索的同时保存节点和节点的深度。当第一次遇到叶节点时,则结束。
/**
* Definition for a binary tree node.
* public class 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;
* }
* }
*/
class Solution {
class QueueNode{
TreeNode node;
int depth;
QueueNode(TreeNode n,int d){
node=n;
depth=d;
}
}
public int minDepth(TreeNode root) {
if(root==null) return 0;
Queue queue=new LinkedList<>();
queue.add(new QueueNode(root,1));
while(!queue.isEmpty()){
QueueNode n=queue.poll();
TreeNode node=n.node;
int depth=n.depth;
if(node.left==null&&node.right==null) return depth;
if(node.left!=null) queue.add(new QueueNode(node.left,depth+1));
if(node.right!=null) queue.add(new QueueNode(node.right,depth+1));
}
return 0;
}
}