Binary Search:222. Count Complete Tree Nodes

努力努力再努力r

于 2017-10-23 20:46:13 发布

阅读量187

点赞数

分类专栏： leetcode Binary Search

本文链接：https://blog.youkuaiyun.com/u012528000/article/details/78323110

版权

leetcode 同时被 2 个专栏收录

264 篇文章

订阅专栏

Binary Search

26 篇文章

订阅专栏

本文介绍了一种计算完美二叉树节点数量的方法。利用递归方式比较左子树和右子树高度来判断是否为满二叉树，并据此返回节点总数。适用于算法和数据结构的学习。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

若是完美二叉树，节点个数很好求，为2的h次方-1，h为该完美二叉树的高度。这道题可以用递归和非递归两种方法来解。我们先来看递归的方法，思路是分别找出以当前节点为根节点的左子树和右子树的高度并对比，如果相等，则说明是满二叉树，直接返回节点个数，如果不相等，则节点个数为左子树的节点个数加上右子树的节点个数再加1(根节点)，其中左右子树节点个数的计算可以使用递归来计算。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int countNodes(TreeNode* root) {
        int l = 0, r = 0;
        TreeNode* pleft = root, *pright = root;
        while(pleft){
            l++;
            pleft = pleft->left;
        }
        while(pright){
            r++;
            pright = pright->right;
        }
        if(l == r)  return pow(2, l)-1;
        return countNodes(root->left) + countNodes(root->right) + 1;
    }
};