[剑指offer]求二叉树的深度

最新推荐文章于 2021-02-04 19:36:37 发布

原创最新推荐文章于 2021-02-04 19:36:37 发布 · 219 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer 同时被 2 个专栏收录

16 篇文章

订阅专栏

剑指offer

14 篇文章

订阅专栏

本文介绍了一种求解二叉树深度的方法，包括非递归的层次遍历实现及递归实现方式，帮助理解二叉树的基本操作。

题目

求二叉树的深度

描述

输入一棵二叉树，求该树的深度。从根结点到叶结点依次经过的结点（含根、叶结点）形成树的一条路径，最长路径的长度为树的深度。

思路

通过层次遍历，获取层数即可
递归思路

代码

通过层次遍历，获取层数即可

public class Solution {
    public int TreeDepth(TreeNode pRoot)
    {
        if(pRoot == null){
            return 0;
        }
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        queue.add(pRoot);
        int depth = 0, count = 0, nextCount = 1;
        while(queue.size()!=0){
            TreeNode top = queue.poll();
            count++;
            if(top.left != null){
                queue.add(top.left);
            }
            if(top.right != null){
                queue.add(top.right);
            }
            if(count == nextCount){
                nextCount = queue.size();
                count = 0;
                depth++;
            }
        }
        return depth;
    }
}

递归思想

public class Solution {
    public int TreeDepth(TreeNode pRoot)
    {
        if(pRoot == null){
            return 0;
        }
        int left = TreeDepth(pRoot.left);
        int right = TreeDepth(pRoot.right);
        return Math.max(left, right) + 1;
    }
}