022-leetcode-树-树的最大高度

package tree;

import java.util.Deque;
import java.util.LinkedList;
import java.util.Queue;

/**
 * 二叉树的最大深度
 * 给定一个二叉树，找出其最大深度。
 * <p>
 * 二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。
 * <p>
 * 说明: 叶子节点是指没有子节点的节点。
 * <p>
 * 示例：
 * 给定二叉树 [3,9,20,null,null,15,7]，
 * <p>
 *   3
 * /   \
 * 9    20
 *     /  \
 *   15   7
 * 返回它的最大深度 3 。
 * <p>
 * 作者：力扣 (LeetCode)
 * 链接：https://leetcode-cn.com/leetbook/read/top-interview-questions-easy/xnd69e/
 * 来源：力扣（LeetCode）
 * 著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。
 */
public class MaxDepth {

    private static class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode(int val, TreeNode left, TreeNode right) {
            this.val = val;
            this.left = left;
            this.right = right;
        }

        TreeNode(int val) {
            this.val = val;
        }
    }


    public static void main(String[] args) {
        TreeNode rootNode = new TreeNode(3);
        TreeNode node1 = new TreeNode(9);
        TreeNode node2 = new TreeNode(20);
        TreeNode node3 = new TreeNode(15);
        TreeNode node4 = new TreeNode(7);

        rootNode.left = node1;
        rootNode.right = node2;
        node2.left = node3;
        node2.right = node4;

        int maxDepth = maxDepth(rootNode);
        int maxDepthNew=maxDepthNew(rootNode);
        System.out.println("maxDepth " + maxDepth+" maxDepthNew "+maxDepthNew);
    }

    private static int maxDepth(TreeNode rootNode) {
        return rootNode == null ? 0 : Math.max(maxDepth(rootNode.left), maxDepth(rootNode.right)) + 1;
    }


    // 利用层序遍历的方法 每遍历完一层就层数上加 +1
    private static int maxDepthNew(TreeNode rootNode) {
        if (rootNode == null) {
            return 0;
        }
        Deque<TreeNode> queue = new LinkedList<>();
        queue.addLast(rootNode);
        int count = 0;
        while (!queue.isEmpty()) {
            int size = queue.size();
            while (size-->0) {
                TreeNode node = queue.pop();
                if (node.left != null) {
                    queue.addLast(node.left);
                }
                if (node.right != null) {
                    queue.addLast(node.right);
                }
            }
            count++;
        }
        return count;
    }
}

输出结果：

maxDepth 3 maxDepthNew 3