代码随想录——二叉树-11.完全二叉树的节点个数

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

思路

一、层序遍历,时间复杂度O(n)

二、利用完全二叉树性质,时间复杂度O(logn * logn)(小于O(n))

完全二叉树性质:若树深度为h,则前h-1层节点都达到最大值。第h层节点都集中在最左侧的位置
完全二叉树要么1.是满二叉树 2.最后一层没满

  1. 满二叉树计算节点数太方便了,直接用公式2^h-1。

  2. 最后一层没满则遍历根的左右子树,递归到某一深度一定会有左孩子或者右孩子为满二叉树,然后依然可以按照情况1来计算

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

那么现在的问题就是怎么判断一棵树是否是满二叉树:在完全二叉树中,若左右子树的深度一样则为满二叉树

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int countNodes(TreeNode* root) {
        //1.终止条件
        if(root == nullptr)return 0;
        //2.中间代码
        int lheight=0,rheight=0;
        TreeNode* lnode = root->left,*rnode = root->right;
        while(lnode != nullptr){ //左子树深度
            lheight++;
            lnode = lnode->left;
        }
        while(rnode != nullptr){ //右子树深度
            rheight++;
            rnode = rnode->right;
        }
        if(lheight == rheight){
            return (2 << lheight) - 1; //用位运算代替幂,速度更快
        }
        return countNodes(root->left) + countNodes(root->right) + 1;
    }
};

总结

本题难点在于怎么利用完全二叉树性质,对于满二叉树用公式计算节点很容易想到,但想到非满二叉树可以递归直到找到有左右子树是满二叉树则不容易。而利用完全二叉树性质判断满二叉树也是第一次见。

此外还要分析清楚递归的三大条件(递归函数,终止条件和单层递归逻辑),树的遍历顺序(左右中,后序遍历)

posted @   NeroMegumi  阅读(27)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 10年+ .NET Coder 心语 ── 封装的思维:从隐藏、稳定开始理解其本质意义
· 地球OL攻略 —— 某应届生求职总结
· 提示词工程——AI应用必不可少的技术
· Open-Sora 2.0 重磅开源!
· 周边上新:园子的第一款马克杯温暖上架
点击右上角即可分享
微信分享提示