leetcode222. 完全二叉树的节点个数(两种做法)

最新推荐文章于 2024-10-02 10:19:33 发布

天天向上的菜鸡杰！！

最新推荐文章于 2024-10-02 10:19:33 发布

阅读量184

点赞数

分类专栏： 101道leetcode 寒假算法训练营算法分析与设计文章标签：深度优先数据结构 c++

本文链接：https://blog.csdn.net/qq_48508278/article/details/122484846

版权

算法分析与设计同时被 3 个专栏收录

224 篇文章 11 订阅

订阅专栏

寒假算法训练营

144 篇文章 1 订阅

订阅专栏

101道leetcode

110 篇文章 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:

    void preorder(TreeNode* root,vector<int>&v) {
        if(root == NULL) return;
        v.push_back(root->val);
        preorder(root->left,v);
        preorder(root->right,v);
    }

    int countNodes(TreeNode* root) {
        /**
        思路:这个也是我们遍历二叉树,采用一种遍历方式去统计其的个数（层序可以,先序也行）
        */
        vector<int> ans;
        preorder(root,ans);
        return ans.size();
    }
};

方法二：递归

/**
 * 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
                int getNums(TreeNode*root)
        第二步:确定递归函数的终止条件
                当节点遇到NULL就停止
                if(root == NULL) return 0;(0代表没有个数)
        第三步:确定单层递归逻辑
                我们遍历左子树的节点数，再遍历右子树的节点数,然后再将左右节点数相加 再加1
                (这个1代表的是中间的节点）                

    */
    int getNums(TreeNode* root){
        if(root == NULL) return 0;
        int leftNums = getNums(root->left);//左子树的个数 （左子树当中也会有左右子树 来统计个数）
        int rightNums = getNums(root->right);//右子树的个数
        int ans = leftNums+rightNums+1;//将左右子树的个数相加 + 中间的节点

        return ans;
    }

    int countNodes(TreeNode* root) {
        
        int ans = getNums(root);
        return ans;
    }
};