Java实现求二叉树的深度和宽度

面试常问题目,总结一下,这个是常见的对二叉树的操作。

解释:

二叉树的深度:从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
二叉树的宽度:二叉树的每一层中都有一定数量的节点,节点数最多的那一层的节点数叫做二叉树的宽度。

设节点的数据结构,如下:

class TreeNode {
    char val;
    TreeNode left = null;
    TreeNode right = null;
    TreeNode(char val) {
        this.val = val;
    }
}

1.二叉树深度

思路:递归实现。

  1. 每个节点都可以看作根节点
  2. 根节点(任意一个节点)的深度等于它的左子树或右子树深度最大值+1
  3. 从根结点开始遍历,若遍历到叶子节点,深度为0

使用递归,分别求出左子树的深度、右子树的深度,两个深度的较大值+1即可。代码如下:

public static int getMaxDepth(TreeNode root) {
        if (root == null) {
            return 0;
        }else {
            int left = getMaxDepth(root.left);
            int right = getMaxDepth(root.right);
            return 1 + Math.max(left, right);
        }
    }

2.二叉树宽度

思路:层序遍历时添加一个计数器,记录每层的节点数

  1. 每层出队列时记录下一层的节点数,其实就是队列的Size()
  2. 每层遍历结束时,比较最大宽度与当前层节点数,记录最大值

使用队列,层次遍历二叉树。在上一层遍历完成后,下一层的所有节点已经放到队列中,此时队列中的元素个数就是下一层的宽度。以此类推,依次遍历下一层即可求出二叉树的最大宽度。代码如下:

public static int getMaxWidth(TreeNode root) {
        if (root == null) { return 0; }
        Queue<TreeNode> queue = new ArrayDeque<TreeNode>();
        int maxWidth = -1; // 最大寬度
        queue.add(root);   // 入队
        while (true) {
            int len = queue.size(); // 当前层的节点个数
            if (len == 0) {
                break;
            }
            while (len > 0) { // 当前层还有节点
                TreeNode t = queue.poll(); // 从队列头部删除一个元素。
// 在队列元素为空的情况下,remove() 方法会抛出NoSuchElementException异常,poll() 方法只会返回 null。
//(remove() 和 poll()有什么区别?)
                len--;
                if (t.left != null) {
                    queue.add(t.left); // 下一层节点入队
                }
                if (t.right != null) {
                    queue.add(t.right); // 下一层节点入队
                }
            }
            maxWidth = Math.max(maxWidth, queue.size());
        }
        return maxWidth;
    }

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值