求二叉树的深度和宽度

最新推荐文章于 2018-08-16 11:05:00 发布

原创最新推荐文章于 2018-08-16 11:05:00 发布 · 1k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树 #算法面试题

经典算法面试题专栏收录该内容

51 篇文章

订阅专栏

本文介绍了一种计算二叉树宽度和深度的方法。通过递归方式计算二叉树的高度，采用队列实现来确定二叉树的最大宽度。适用于数据结构与算法的学习者。

题目描述：求二叉树的宽度和深度
给定一个二叉树，获取该二叉树的宽度和深度。

    //二叉树的高度：
    public static int getHeight(BiNode head){
        if(head == null){
            return 0;
        }
        int leftHeight = getHeight(head.left);
        int rightHeight = getHeight(head.right);
        if(leftHeight > rightHeight){
            return leftHeight + 1;
        }
        return rightHeight + 1;
    }

    //二叉树的宽度：
    public static int getWidth(BiNode head){
        if(head == null){
            return 0;
        }
        Queue<BiNode> queue = new ArrayDeque<BiNode>();
        int maxWidth = 1;
        queue.add(head);
        while(true){
            int len = queue.size();
            if(len == 0)
                break;
            while(len > 0){
                BiNode temp = queue.poll();
                len--;
                if(temp.left != null){
                    queue.add(temp.left);
                }
                if(temp.right != null){
                    queue.add(temp.right);
                }
            }
            maxWidth = Math.max(maxWidth, queue.size());
        }
        return maxWidth;
    }