在这里插入图片描述

递归法

1、确定递归函数的参数和返回值:参数就是传⼊树的根节点,返回就返回以该节点为根节点⼆叉树的节点数量,所以返回值为int类型。

int countNodes(TreeNoed* root)

2、确定终⽌条件:如果为空节点的话,就返回0,表示节点数为0。代码如下:

if(node==nullptr) return 0;

3、确定单层递归的逻辑:先求它的左⼦树的节点数量,再求的右⼦树的节点数量,最后取总和再加⼀(加1是因为算上当前中间节点)就是⽬前节点为根节点的节点数量。

 int leftnodes=getnodes(node->left); //左
 int rightnodes=getnodes(node->right);//右
 int sum=leftnodes+rightnodes;//中
 return sum+1;

完整递归代码:

class Solution {
public:
    int getnodes(TreeNode* node)
    {
      if(node==nullptr) return 0;
      int leftnodes=getnodes(node->left); //左
      int rightnodes=getnodes(node->right);//右
      int sum=leftnodes+rightnodes;//中
      return sum+1;
    }
    
    int countNodes(TreeNode* root) {
        return getnodes(root);
    }
};

时间复杂度:O(n)
空间复杂度:O(logn),算上了递归系统栈占⽤的空间

迭代法

class Solution{
public:
    int countNodes(TreeNode* root){
       if(root==nullptr) return 0;
       queue<TreeNode*> que;
       que.push(root);
       int sum=0;
       
       while(!que.empty()){
         int size=que.size();
         for(int ii=0;ii<size;ii++){
           sum++;
           TreeNode* node=que.front();
           que.pop();
           if(node->left) que.push(node->left);
           if(node->right) que.push(node->right);
         }
       }
       return sum;
    }
};

时间复杂度:O(n)空间复杂度:O(n)

利用完全二叉树的性质

完全二叉树只有两种情况:
1、满二叉树—–>高度为h的满二叉树有 2^h-1 个结点。
2、最后⼀层叶⼦节点没有满。—->也就是说如果整个树不是满二叉树就分别递归其左右孩子,递归到一定深度一定会有左孩子或右孩子为满二叉树,然后可以按照满二叉树来计算节点个数。

class Solution{
public:
    int countNodes(TreeNode* root){
        if(root==nullptr) return 0;
        
        TreeNode* left=root->left;
        TreeNode* right=root->right;
        
        int leftHeight=0,rightHight=0;
        while(left){ //求左子树的高度
          left=left->left; leftHeight++;
        }
        while(right){ //求右子树的高度
          right=right->right; rightHeight++;
        }
        
        if(leftHeight==rightHeight){
          return (2<<leftHeight)-1; //2<<相当于2^2 ,所以高度初始化为0 
        }

        //如果整棵树不是满二叉树,就递归其左右孩子
        return countNodes(root->left)+countNodes(root->right)+1}
};

时间复杂度:O(logn * logn)
空间复杂度:O(logn)


版权声明:本文为weixin_46269257原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
原文链接:https://blog.csdn.net/weixin_46269257/article/details/115816399