198.完全二叉树的节点个数

给出一个完全二叉树,求出该树的节点个数。
说明:
完全二叉树的定义如下:在完全二叉树中,除了最底层节点可能没填满外,其余每层节点数都达到最大值,并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层,则该层包含 1~ 2h 个节点。

示例:

输入: 
    1
   / \
  2   3
 / \  /
4  5 6
输出: 6

代码

class Solution {
public:
    int countNodes(TreeNode* root) {
        int hLeft = 0, hRight = 0;
        TreeNode *pLeft = root, *pRight = root;
        while (pLeft) {
            ++hLeft;
            pLeft = pLeft->left;
        }
        while (pRight) {
            ++hRight;
            pRight = pRight->right;
        }
        if (hLeft == hRight) return pow(2, hLeft) - 1;
        return countNodes(root->left) + countNodes(root->right) + 1;
    }
};
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 树的概述 树是一种非常常用的数据结构,树与前面介绍的线性表,栈,队列等线性结构不同,树是一种非线性结构 1.树的定...
    Jack921阅读 9,942评论 1 31
  • 前言 树是数据结构中的重中之重,尤其以各类二叉树为学习的难点。一直以来,对于树的掌握都是模棱两可的状态,现在希望通...
    MrHorse1992阅读 354,709评论 51 536
  • 苹果自X-Code6之后创建类默认取消了类前缀。 下面介绍如何添加类前缀:选中“TARGETS”在右侧栏中的“Pr...
    遛遛食阅读 3,790评论 0 0
  • 如果我们制作一个计算器程序,当用户输入的除数为0时,程序将会崩溃直接退出,那么该程序的用户体验将会非常差。我们应该...
    yuluo阅读 3,663评论 0 1
  • 1遇事冷静 2简单 3平凡 4易处 5自尊 6敬人 7保持心情舒畅 8多肯定赞扬亲友 9享受生命而不是抱怨生活 1...
    狮女柔心_Nicole陶阅读 1,033评论 1 2