230. Kth Smallest Element in a BST

最新推荐文章于 2021-04-09 14:18:22 发布

原创最新推荐文章于 2021-04-09 14:18:22 发布 · 252 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 同时被 2 个专栏收录

108 篇文章

订阅专栏

Leetcode_tree

15 篇文章

订阅专栏

本文介绍了一种利用递归与排序方法求解二叉搜索树中第K小元素的问题。通过定义一个辅助函数进行节点遍历并将节点值收集到一个向量中，然后对向量进行排序，从而找到第K小的元素。

题目描述【Leetcode】

Given a binary search tree, write a function kthSmallest to find the kth smallest element in it.

Note:
You may assume k is always valid, 1 ? k ? BST’s total elements.

找出二叉搜索树的第k小的数，用递归再排序

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */

void f(TreeNode * root, vector<int>& re){
    if(!root) return;
    re.push_back(root->val);
    f(root->left,re);
    f(root->right,re);
}
class Solution {
public:
    int kthSmallest(TreeNode* root, int k) {
        if(!root) return 0;
        vector<int>re;
        f(root,re);
        sort(re.begin(),re.end());
        return re[k-1];
    }
};