LEETCODE 49场周赛 第K大完美二叉子树的大小

这个题目本身很直接,谨以此题来熟练一下二叉树的相关操作的使用,好久没写二叉树的代码了,不熟练了有点,题目如下:题目链接传送门在这里题目链接传送门

给你一棵 二叉树 的根节点 root 和一个整数k。

返回第 k 大的 完美二叉子树的大小,如果不存在则返回 -1。

完美二叉树 是指所有叶子节点都在同一层级的树,且每个父节点恰有两个子节点。

子树 是指树中的某一个节点及其所有后代形成的树。

分析:其实简单分析一下哈,这个完美二叉树就是我们的完全二叉树的定义嘛,所以我们只需要判定每个节点是否是完全二叉树,然后返回其高度h,就可以根据这个高度h得到其节点个数为(2 ^h - 1),然后进行排序就好。

具体代码如下:(实现细节详见代码注释)

cpp 复制代码
/**
 * 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:
     //这个函数是去判断该子树是否是完全二叉树的,是的话就返回子树高度,否则返回-1
    int getHeight(TreeNode* root) {
        if (!root) return 0;
        int leftHeight = getHeight(root->left);
        int rightHeight = getHeight(root->right);
        if (leftHeight == -1 || rightHeight == -1 || leftHeight != rightHeight) {
            return -1; // 不是完美二叉树
        }
        return leftHeight + 1; // 是完美二叉树,高度加1
    }

    // 递归函数来计算完美二叉子树的大小,根据height返回 完全二叉树的高度
    int getPerfectSubtreeSize(TreeNode* root) {
        int height = getHeight(root);
        if (height == -1) {
            return 0; // 不是完美二叉树,大小为0
        }
        // 完美二叉树的节点总数为 2^height - 1
        return (1 << height) - 1;
    }

    // 主函数
    int kthLargestPerfectSubtree(TreeNode* root, int k) {
        vector<int> sizes;
        findPerfectSubtrees(root, sizes);
        // 对所有完美二叉子树的大小进行排序
        sort(sizes.rbegin(), sizes.rend());
        // 返回第 k 大的完美二叉子树的大小
        return (k <= sizes.size()) ? sizes[k - 1] : -1;
    }

    private:
    // 辅助函数,递归查找所有完美二叉子树,遍历整个二叉树,采用的是前序遍历,把所有合法节点的节点数目保存到sizes数组里面
        void findPerfectSubtrees(TreeNode* root, vector<int>& sizes) {
            if (!root) return;
            int size = getPerfectSubtreeSize(root);
            if (size > 0) {
                sizes.push_back(size);
            }
            // 继续递归查找左右子树
            findPerfectSubtrees(root->left, sizes);
            findPerfectSubtrees(root->right, sizes);
        }

};
相关推荐
阿维的博客日记1 小时前
LeetCode 139. 单词拆分 - 动态规划解法详解
leetcode·动态规划·代理模式
程序员Xu2 小时前
【LeetCode热题100道笔记】二叉树的右视图
笔记·算法·leetcode
笑脸惹桃花2 小时前
50系显卡训练深度学习YOLO等算法报错的解决方法
深度学习·算法·yolo·torch·cuda
阿维的博客日记3 小时前
LeetCode 48 - 旋转图像算法详解(全网最优雅的Java算法
算法·leetcode
GEO_YScsn3 小时前
Rust 的生命周期与借用检查:安全性深度保障的基石
网络·算法
程序员Xu3 小时前
【LeetCode热题100道笔记】二叉搜索树中第 K 小的元素
笔记·算法·leetcode
THMAIL4 小时前
机器学习从入门到精通 - 数据预处理实战秘籍:清洗、转换与特征工程入门
人工智能·python·算法·机器学习·数据挖掘·逻辑回归
Kevinhbr4 小时前
CSP-J/S IS COMING
数据结构·c++·算法
蕓晨5 小时前
set的插入和pair的用法
c++·算法
THMAIL5 小时前
深度学习从入门到精通 - AutoML与神经网络搜索(NAS):自动化模型设计未来
人工智能·python·深度学习·神经网络·算法·机器学习·逻辑回归