力扣LeetCode经典算法 二叉搜索树的第k大节点

数据结构(六十四)

学习数据结构与算法过程中的心得体会以及知识点的整理,方便我自己查找,也希望可以和大家一起交流。

—— 二叉搜索树的第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大的数字,我们还需要将中序遍历的序列逆置,那么为了方便,我们在中序遍历的时候直接一步到位,把遍历顺序从“左中右”变成“右中左”,这样得出来的序列就是降序的序列了。

你可能感兴趣的:(数据结构与算法,leetcode,算法,数据结构,二叉树)