一、题目
网址:二叉搜索树中第K小的元素。
级别:中等。
给定一个二叉搜索树,编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。
说明:
你可以假设 k 总是有效的,1 ≤ k ≤ 二叉搜索树元素个数。
示例 1:
输入: root = [3,1,4,null,2], k = 1
3
/ \
1 4
\
2
输出: 1
示例 2:
输入: root = [5,3,6,2,4,null,null,1], k = 3
5
/ \
3 6
/ \
2 4
/
1
输出: 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:
vector<int> num;
//中序遍历,将结果存入num中
void preOrder(TreeNode *root)
{
if(root==NULL) return;
preOrder(root->left);
num.push_back(root->val);
preOrder(root->right);
}
int kthSmallest(TreeNode* root, int k) {
preOrder(root);
return num[k-1];
}
};
结果:
方法二
采用非递归(栈)的形式对二叉搜索树进行中序遍历,得到从小到大排列的数组。
/**
* 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:
int kthSmallest(TreeNode* root, int k) {
vector<int> num;
stack<TreeNode *> s;
while(!s.empty()||root!=NULL)
{
while(root!=NULL)
{
s.push(root);
root=root->left;
}
if(!s.empty())
{
root=s.top();
num.push_back(root->val);
s.pop();
root=root->right;
if(k==num.size())
{
return num[k-1];
}
}
}
return num[k-1];
}
};
结果:
注:在使用方法二时进行了优化,找到第K小的元素即可终止,不用再继续遍历,因此数组num更小,使用的内存更小。