求给定二叉树的最小深度java实现

最小深度是指树的根结点到最近叶子结点的最短路径上结点的数量。

递归思路为:

若为空树返回0;

若左子树为空,则返回右子树的最小深度+1;

若右子树为空,则返回左子树的最小深度+1;

若左右子树均不为空,则取左、右子树最小深度的较小值,+1;

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public int run(TreeNode root) {
        if(root==null)
            return 0;
        if(root.left==null&&root.right!=null)
            return run(root.right)+1;
        if(root.left!=null&&root.right==null)
            return run(root.left)+1;
        return Math.min(run(root.left)+1,run(root.right)+1);
    }
}

非递归思路:利用队列对树进行层序遍历,找到第一个左右节点都为null的子节点,返回当前深度。

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
import java.util.*;
public class Solution {
    public int run(TreeNode root) {
        if(root==null)
            return 0;
        Queue<TreeNode> queue=new LinkedList<TreeNode>();
        queue.add(root);
        int depth=0;
        while(!queue.isEmpty()){
            int count=queue.size();
            depth++;
            while(count>0){
                TreeNode temp=queue.poll();
                if(temp.left==null&&temp.right==null)
                    return depth;
                if(temp.left!=null)
                    queue.add(temp.left);
                if(temp.right!=null)
                    queue.add(temp.right);
                count--;
            }
        }
        return 0;
    }
}
发布了32 篇原创文章 · 获赞 1 · 访问量 357

猜你喜欢

转载自blog.csdn.net/zy854816286/article/details/104798844