辛星Java树算法教程第十二篇:二叉树的最小深度

对于二叉树来说，所谓最小深度，就是树的叶子节点中到根节点的最小距离。
这里说一下，空树返回0.
如果一个二叉树只有右子树，则返回右子树上的叶节点的最小深度。
如果一个二叉树只有左子树，则返回左子树上的叶节点的最小深度。
如果一个二叉树有左右子树，则返回左右子树上的最小深度。

我们当然可以使用递归的方式来做，我们也可以使用分层遍历的方式来解决。
下面给出示例代码:

package com.mengzhidu.teach.algorithm.tree.demo.basic;

import com.mengzhidu.teach.algorithm.tree.demo.TreeNode;
import com.mengzhidu.teach.algorithm.tree.demo.TreeNodeHelper;

import java.util.ArrayDeque;
import java.util.Queue;

/**
 * 判断树的最小深度
 * 所谓树的最小深度，即树的叶子节中距离根节点的最小距离
 * 这里我们可以采用分层遍历，然后我们找到第一个叶子节点即可
 */
public class MinDepthDemo {

    public static void main(String[] args) {
        TreeNode root = TreeNodeHelper.getTreeNode();
        System.out.println("树的最小深度:" + getMinHeight(root));
    }

    private static int getMinHeight(TreeNode node) {
        if (node == null) {
            return 0;
        }

        Queue<TreeNode> queue = new ArrayDeque<>();
        queue.add(node);

        int height = 0;
        while (!queue.isEmpty()) {
            int width = queue.size();
            if (width == 0) {
                break;
            }
            height ++;
            while (width > 0) {
                TreeNode tmp = queue.poll();
                if (tmp.getLeft() == null && tmp.getRight() == null) {
                    return height;
                }
                width --;
                if (tmp.getLeft() != null) {
                    queue.add(tmp.getLeft());
                }
                if (tmp.getRight() != null) {
                    queue.add(tmp.getRight());
                }
            }
        }

        return height;
    }
}

对于二叉树的最小深度，我们就介绍到这里啦。