标签：Info head return int 个数 二叉树 节点

作者：Grey

原文地址：快速求完全二叉树的节点个数

题目链接

LeetCode 222. 完全二叉树的节点个数

题目进阶要求

进阶：遍历树来统计节点是一种时间复杂度为 O(n) 的简单解决方案。你可以设计一个更快的算法吗？

暴力解法

不考虑完全二叉树的这个性质，直接遍历一下二叉树，收集一下左右子树的节点个数，然后加上头节点，就是整个完全二叉树的节点个数，完整代码如下

    public static int countNodes1(TreeNode head) {
        if (head == null) {
            return 0;
        }
        return p(head).all;
    }
    public static Info p(TreeNode head) {
        if (head == null) {
            return new Info(0);
        }
        Info leftInfo = p(head.left);
        Info rightInfo = p(head.right);
        // 收集左右子树节点个数加上头节点，就是整棵树的节点个数
        int all = leftInfo.all + rightInfo.all + 1;
        return new Info(all);
    }

    public static class Info {
        public int all;
        public Info(int a) {
            all = a;
        }
    }

时间复杂度是O(N)。

最优解

需要利用一下完全二叉树的性质，首先，我们知道，树的高度，深度，层次有如下关系：

对于一棵完全二叉树，头节点一直往左滑直到叶子节点得到的层数一定是这个二叉树的最大层数，假设这个值为h。

如果这个二叉树右树的最左节点的层数正好等于h，说明这个二叉树左树一定是满二叉树

如果这个二叉树右树的最左节点的层数不等于h，则说明这个二叉树的右树一定是满二叉树

由于满二叉树的节点个数可以通过树的高度计算出来：

    public static int countNodes(TreeNode head) {
        if (head == null) {
            return 0;
        }
        int h = maxLenOfLeft(head, 1);
        return count(head, 1, h);
    }

    private static int count(TreeNode head, int level, int h) {
        if (level == h) {
            return 1;
        }
        if (maxLenOfLeft(head.right, level + 1) == h) {
            // 左树一定是满的
            return (1 << (h - level)) + count(head.right, level + 1, h);
        } else {
            // 右数一定是满的，注意高度是 h - level - 1
            return (1 << (h - level - 1)) + count(head.left, level + 1, h);
        }
    }

    public static int maxLenOfLeft(TreeNode root, int level) {
        while (root != null) {
            root = root.left;
            level++;
        }
        return level - 1;
    }

通过如上方法，我们无须遍历整个二叉树，就可以把节点计算出来，复杂度低于O(N)。

更多

算法和数据结构笔记

标签：Info,head,return,int,个数,二叉树,节点
来源： https://www.cnblogs.com/greyzeng/p/16360786.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。