230. Kth Smallest Element in a BST

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

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

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

11 篇文章

订阅专栏

本文介绍了一种解决二叉搜索树中寻找第K小元素的方法。通过宽度优先遍历获取树中所有元素，并对其进行排序来找到目标元素。文章提供了具体的实现代码。

1. 原题

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 element.

2. 分析

题目要求找出宽度优先树中的第K小的树。
在没有这棵树的数据的情况下不知道哪些数是存在的，所以要确定第K个树，必须要遍历一遍树。得到所有数之后排序就可以得到第K小的数了
宽度优先和深度优先的遍历都有规律可循，分别判断根，坐子树和右子树，基本普通的遍历都是这个模式，需要考虑的是退出情况还有传参的方式，这里使用引用传参比较方便

 if (root != NULL){
            nums.push_back(root->val);
        }
        if (root->left != NULL) getNums(root->left, nums);
        if (root->right != NULL) getNums(root->right, nums);
}

3.代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void getNums(TreeNode* root, vector<int>& nums){
        if (root != NULL){
            nums.push_back(root->val);
        }
        if (root->left != NULL) getNums(root->left, nums);
        if (root->right != NULL) getNums(root->right, nums);
    }
    int kthSmallest(TreeNode* root, int k) {
        vector<int> nums;
        int len = nums.size();
        getNums(root, nums);
        sort(nums.begin(), nums.end());
        return nums[k-1];
    }
};