[M二叉树] lc222. 完全二叉树的节点个数(二叉树+二分查找)

最新推荐文章于 2024-04-06 10:52:47 发布

Ypuyu

最新推荐文章于 2024-04-06 10:52:47 发布

阅读量183

点赞数

分类专栏： LeetCode 文章标签：二叉树二分

本文链接：https://blog.csdn.net/yl_puyu/article/details/110091061

版权

LeetCode 专栏收录该内容

385 篇文章 1 订阅

订阅专栏

本文探讨了如何使用广度优先搜索（BFS）、深度优先搜索（DFS）以及二分法解决LeetCode 222题——完全二叉树节点个数问题。通过实例代码展示了三种算法的实现，并介绍了完全二叉树的性质在优化计数过程中的作用。

摘要由CSDN通过智能技术生成

文章目录

1. 题目来源

链接：lc222. 完全二叉树的节点个数

2. 题目说明

在这里插入图片描述

3. 题目解析

方法一：bfs

。。。

代码：

/**
 * 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) return 0;
        int res = 1;
        queue<TreeNode*> q;
        q.push(root);
        while (q.size()) {
            int len = q.size();
            for (int i = 0; i < len; ++i) {
                auto t = q.front();
                q.pop();
                if (t->left) q.push(t->left), ++res;
                if (t->right) q.push(t->right), ++res;
            }
        }
        return res;
    }
};

方法二：dfs

。。。

代码：

class Solution {
public:
    int countNodes(TreeNode* root) {
        if (!root) return 0;
        return countNodes(root->left) + countNodes(root->right) + 1;
    }
};

方法二：二分

首先完全二叉树和满二叉树的性质请自行百度。

思路：

是满二叉树： 在完全二叉树基础上，如果从根节点一直走左孩子得到树高度，和从根节点一直走右孩子得到树高度一样的话，该完全二叉树就是一颗满二叉树。那么直接返回 $2^n-1$ 即可
不是满二叉树： 有两种情况：
- 若左子树和右子树都是满二叉树，则直接计算即可
- 若左子树和右子树其中之一不是满二叉树，则计算满二叉树一边，另一边递归处理即可
- 注意： 左右子树都是满二叉树的话，那么最后一层节点分界点就刚好在中间位置。否则，分界点在左侧、右侧，都会造成左子树、右子树不为满二叉树的情况

画图很好理解。

树上的二分，挺不错的一道题目。

总共二分 $l o g n$ 次，每次判断当前树高度，是 $l o g n$ ，所有总的时间复杂度是 $O(log^2_n)$

代码：

/**
 * 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) return 0;
        auto l = root->left, r = root->right;
        int x = 1, y = 1;
        while (l) l = l->left, x ++;
        while (r) r = r->right, y ++;
        if (x == y) return (1 << x) - 1;
        return countNodes(root->left) + 1 + countNodes(root->right);
    }
};