树-求最大深度的层序遍历解法

最新推荐文章于 2023-07-27 12:45:38 发布

原创最新推荐文章于 2023-07-27 12:45:38 发布 · 386 阅读

0 ·

CC 4.0 BY-SA版权

数据结构与算法-树专栏收录该内容

13 篇文章

订阅专栏

本文介绍了一种计算二叉树最大深度的方法——层序遍历。通过使用队列来跟踪每一层的节点，该算法能有效地遍历整棵树并返回其深度。示例代码演示了如何创建一棵二叉树并调用maxDepth方法获取树的最大深度。

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

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 maxDepth(TreeNode root){
        int depth = 0;
        Queue<TreeNode> queueNode = new LinkedList<>();
        queueNode.add(root);
        while (!queueNode.isEmpty()){
            for (int i = 0; i < queueNode.size(); i++) {
                if(queueNode.peek().left != null)queueNode.add(queueNode.peek().left);
                if(queueNode.peek().right != null)queueNode.add(queueNode.peek().right);
                queueNode.poll();
            }
            depth++;
        }
        return depth;
    }
}

class test{
    public static void main(String[] args) {
        TreeNode node1 = new TreeNode(1);
        TreeNode node2 = new TreeNode(2);
        TreeNode node3 = new TreeNode(3);
        TreeNode node4 = new TreeNode(4);
        TreeNode node5 = new TreeNode(5);
        TreeNode node6 = new TreeNode(6);
        TreeNode node7 = new TreeNode(7);
        node1.left = node2;node1.right = node3;
        node2.left = node4;node2.right = node5;
        node3.left = node6;node3.right = node7;
        System.out.println(new Solution().maxDepth(node1));
    }
}