222. Count Complete Tree Nodes

最新推荐文章于 2024-01-20 11:36:34 发布

zeroQiaoba

最新推荐文章于 2024-01-20 11:36:34 发布

阅读量354

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/zeroQiaoba/article/details/62227260

leetcode 专栏收录该内容

429 篇文章

订阅专栏

本文介绍了一种优化的二叉树节点计数方法，通过递归算法提高效率，避免了采用对分查找时可能出现的超时问题，并提供了详细的实现代码。

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

答案正确但是超时，==，本以为采用对分的方法就可以了，，哭==，帖上对分的方式。

/**
 * 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:

    bool judge(int num,int height,TreeNode* root)
    {
        stack<int> path;
        for(int i=0;i<height+1;i++)
        {
            path.push(num);
            num=num/2;
        }

        TreeNode* temp=root;

        int top=path.top();
        path.pop();
        while((!path.empty())&&temp!=NULL)
        {
            if(path.top()-top*2==0)
                temp=temp->left;
            else
                temp=temp->right;
            top=path.top();
            path.pop();
        }

        if(temp==NULL)
            return false;
        else
            return true;
    }

    int countNodes(TreeNode* root) {

        if(root==NULL)
            return 0;

        //get height
        int height=0;
        TreeNode* temp=root;
        while(!(temp->left==NULL&&temp->right==NULL))
        {
            temp=temp->left;
            height++;
        }


        int left=0;
        int right=pow(2,height)-1;

        while(right-left>1)
        {
            int mid=left+(right-left)/2;
            if(judge(mid,height,root)==true)
                left=mid;
            else
                right=mid-1;
        }
        //cout<<height;

        if(judge(right,height,root)==true)
            return right+1+pow(2,height)-1;
        else if(judge(left,height,root)==true)
            return left+1+pow(2,height)-1;
        else{}
        return 0;

    }
};

参考了discuss，它采用的是递归的方法，参考其方法求解。最终AC代码，发现运行时间缩短了好多。

/**
 * 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) {

        if(root==NULL)
            return 0;

        int left=0;
        int right=0;

        TreeNode* temp=root;
        while(temp!=NULL)
        {
            temp=temp->left;
            left++;
        }

        temp=root;
        while(temp!=NULL)
        {
            temp=temp->right;
            right++;
        }

        if(left==right)
            return pow(2,left)-1;
        else
        {
            return 1+countNodes(root->left)+countNodes(root->right);
        }

    }
};