[leetcode]230. 二叉搜索树中第K小的元素

最新推荐文章于 2025-06-08 16:34:27 发布

原创最新推荐文章于 2025-06-08 16:34:27 发布 · 195 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种在二叉搜索树中查找第k个最小元素的方法，通过递归算法结合二叉排序树的特性，实现了一个高效的kthSmallest函数。该函数首先计算左子树的节点数，以此判断目标节点的位置，从而实现快速查找。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.题目：
给定一个二叉搜索树，编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。
说明：
你可以假设 k 总是有效的，1 ≤ k ≤ 二叉搜索树元素个数。

示例 :
输入: root = [5,3,6,2,4,null,null,1], k = 3
       5
      / \
     3   6
    / \
   2   4
  /
 1
输出: 3

2.代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */

int getNodeNum(struct TreeNode* root) {
    if(!root)
        return 0;
    return getNodeNum(root->left)+getNodeNum(root->right)+1;
}

//第k则左子树要有k-1个节点。若小与k-1个节点则在右边，否则在左边。
//往左走都一样，还是找第k个节点。
//往右走则是是找第k-num-1个节点，'1'是根节点。

int kthSmallest(struct TreeNode* root, int k) {
    if(!root)
        return -1;
    int leftNums=getNodeNum(root->left);
    if(leftNums==k-1)                	   //中
        return root->val;
    else if(leftNums>k-1)            	   //还在左边
        return kthSmallest(root->left, k);
    else                              	   //在右边
        return kthSmallest(root->right, k-leftNums-1);    
}

3.知识点：

递归，二叉排序树