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

最新推荐文章于 2024-05-23 20:59:40 发布

kelianlee

最新推荐文章于 2024-05-23 20:59:40 发布

阅读量168

点赞数

本文链接：https://blog.youkuaiyun.com/leekerian/article/details/104430068

版权

Leetcode 同时被 2 个专栏收录

73 篇文章

订阅专栏

复习

65 篇文章

订阅专栏

本文介绍了一种高效的算法，用于计算二叉树中节点的数量。通过递归地确定树的深度，并根据深度和遍历策略来计算节点总数，避免了对每个节点进行单独计数的传统方法。

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

/*
    最底层的中间位置为根节点右儿子一直往左边遍历的点
    我们每次判断这个点是否为空如果为空那就在左边否则在右边
    
*/
class Solution {
public:
    int countNodes(TreeNode* root) {
        if(!root) return 0;
        int depth = f(root);
        int ans=(1<<depth)- 1;
        int tmp=1<<(depth-1);
        for(int i=1;i<=depth;i++)
        {
            tmp/=2;
            if(f(root->right)+i==depth) root=root->right;
            else 
            {
                ans-=tmp;
                root=root->left;
            }
        }
        return ans;
    }
    int f(TreeNode *root)
    {
        if(root == NULL) return 0;
        return f(root->left)+1;
    }
};