数据结构(六十四)
学习数据结构与算法过程中的心得体会以及知识点的整理,方便我自己查找,也希望可以和大家一起交流。
—— 二叉搜索树的第k大节点 ——
1.题目描述
给定一棵二叉搜索树,请找出其中第 k 大的节点的值。
示例
输入:
root = [3,1,4,null,2], k = 1
3
/ \
1 4
\
2
输出:
4
输入:
root = [5,3,6,2,4,null,null,1], k = 3
5
/ \
3 6
/ \
2 4
/
1
输出:
4
2.代码
c
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
void ldr(struct TreeNode* root, int k, int *returnSize, int res[]){
if(root!=NULL){
ldr(root->right,k,returnSize,res);
res[(*returnSize)++]=root->val;
ldr(root->left,k,returnSize,res);
}
}
int kthLargest(struct TreeNode* root, int k){
int *res=(int *)malloc(sizeof(int)*10000);
int returnSize=0;
ldr(root,k,&returnSize,res);
return res[k-1];
}
因为二叉搜索树的中序遍历的递增序列,同时为了方便进行寻找第K大的数字,我们还需要将中序遍历的序列逆置,那么为了方便,我们在中序遍历的时候直接一步到位,把遍历顺序从“左中右”变成“右中左”,这样得出来的序列就是降序的序列了。