力扣 222题 完全二叉树的节点个数 记录

题目描述

给你一棵 完全二叉树 的根节点 root ，求出该树的节点个数。

完全二叉树 的定义如下：在完全二叉树中，除了最底层节点可能没填满外，其余每层节点数都达到最大值，并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层，则该层包含 1~ 2h 个节点。

示例1：
       1
      / \
     2   3
    / \ / 
   4  5 6  
   
输入：root = [1,2,3,4,5,6]
输出：6

示例 2：
输入：root = []
输出：0

示例 3：
输入：root = [1]
输出：1

思路

利用完全二叉树的性质来解决这道题，对于一个完全二叉树：

• 如果左子树的高度等于右子树的高度，则左子树是满二叉树；（这是因为如果左子树高度等于右子树高度，说明左子树和右子树的层数是相同的，而由于完全二叉树的定义，除了最后一层外，所有层的节点都是满的。因此左子树是满二叉树。）
• 如果左子树的高度大于右子树的高度，则右子树是满二叉树。
• 

完整代码

#include<iostream>
#include<queue>

using namespace std;

struct TreeNode{
    int val;
    TreeNode* left;
    TreeNode* right;
    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) {
//         if(root == NULL) return 0;
//         int leftNum = countNodes(root->left); // 左
//         int rightNum = countNodes(root->right); // 右
//         int result = leftNum + rightNum + 1; // 中
//         return result;
//         // return countNodes(root->left) + countNodes(root->right) + 1;
//     }
// };

// 层次遍历方法
// class Solution {
// public:
//     int countNodes(TreeNode* root) {
//         queue<TreeNode*> que;
//         if(root != NULL) que.push(root);
//         int count = 0;
//         while(!que.empty()){
//             int size = que.size();
//             for(int i = 0; i < size; i++){
//                 TreeNode* node = que.front();
//                 que.pop();
//                 count ++;
//                 if(node->left) que.push(node->left);
//                 if(node->right) que.push(node->right);
//             }
//         }
//         return count;
//     }
// };

// 利用完全二叉树特性的方法
class Solution{
public:
    int countNodes(TreeNode* root){
        if(root == NULL) return 0;
        TreeNode* left = root->left;
        TreeNode* right = root->right;
        int leftCount = 0, rightCount = 0;
        while(left){
            left = left->left;
            leftCount++;
        }
        while(right){
            right = right->right;
            rightCount++;
        }
        if(leftCount == rightCount){
            return (2 << leftCount) - 1;
        }
        return countNodes(root->left) + countNodes(root->right) + 1;
    }
};

int main()
{
    TreeNode* root = new TreeNode(1);
    root->left = new TreeNode(2);
    root->left->left = new TreeNode(4);
    root->left->right = new TreeNode(5);
    root->right = new TreeNode(3);
    root->right->left =  new TreeNode(6);

    Solution s;
    cout << s.countNodes(root) << endl;

    return 0;
}