leetcode面试题#54之【二叉搜索树的第k大节点】C

最新推荐文章于 2022-04-07 23:30:53 发布

原创最新推荐文章于 2022-04-07 23:30:53 发布 · 266 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #数据结构 #二叉树

leetcode (C) 专栏收录该内容

7 篇文章

订阅专栏

本文介绍了一种在二叉搜索树中查找第k大节点的算法，通过逆向中序遍历并使用辅助数组存储节点数值，当遍历到第k个节点时返回其数值。

题目描述

给定一棵二叉搜索树，请找出其中第k大的节点。
在这里插入图片描述

分析

逆向中序遍历二叉搜索树，使用辅助数组存储遍历节点数值，遍历k个终止递归返回节点数值大小。

代码实现以及执行结果

void inorder(struct TreeNode* root, int* nums, int* returnSize, int k){
    if(!root) return;

    inorder(root->right, nums, returnSize, k);

    nums[(*returnSize)++] = root->val;
    if(returnSize==k) return;

    inorder(root->left, nums, returnSize, k);
}


int kthLargest(struct TreeNode* root, int k){
    if(!root) return NULL;

    int* nums = (int *)malloc(sizeof(int) * 10000);
    int* returnSize = (int *)malloc(sizeof(int));
    *returnSize = 0;
    inorder(root, nums, returnSize, k);

    return nums[k-1];
}